mirror of
https://github.com/thib8956/advent-of-code.git
synced 2024-12-27 06:16:30 +00:00
37 lines
867 B
Python
37 lines
867 B
Python
from itertools import pairwise
|
|
|
|
|
|
def is_safe(report):
|
|
diffs = [b - a for a, b in pairwise(report)]
|
|
return all(1 <= d <= 3 for d in diffs) or all(-3 <= d <= -1 for d in diffs)
|
|
|
|
|
|
def is_safe_with_problem_damper(report):
|
|
for i, _ in enumerate(report):
|
|
copy = report[::]
|
|
del copy[i]
|
|
if is_safe(copy):
|
|
return True
|
|
return False
|
|
|
|
|
|
def main(reports):
|
|
safe_reports = sum(1 for x in reports if is_safe(x))
|
|
print("Part 1: ", safe_reports)
|
|
|
|
safe_reports = 0
|
|
for report in reports:
|
|
if is_safe_with_problem_damper(report):
|
|
safe_reports += 1
|
|
print("Part 2: ", safe_reports)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import sys
|
|
infile = sys.argv[1]
|
|
with open(infile) as f:
|
|
lines = f.readlines()
|
|
lines = [list(map(int, x.rstrip().split())) for x in lines]
|
|
main(lines)
|
|
|