import 2024 aoc

2024/day2/day2.py

@@ -0,0 +1,36 @@
+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)
+