2024: d02: ex2: add solution

2024/d02/ex2/ex2.py

@@ -0,0 +1,36 @@
+#!/usr/bin/env python
+
+import itertools
+import sys
+
+Report = list[int]
+
+
+def solve(input: list[str]) -> int:
+    def parse(input: list[str]) -> list[Report]:
+        return [[int(n) for n in line.split()] for line in input]
+
+    def is_safe(report: Report) -> bool:
+        def is_increasing_safe(report: Report) -> bool:
+            return all(1 <= (b - a) <= 3 for a, b in itertools.pairwise(report))
+
+        return is_increasing_safe(report) or is_increasing_safe(report[::-1])
+
+    # Dampening by brute force
+    def is_dampened_safe(report: Report) -> bool:
+        for i in range(len(report)):
+            dampened_report = report[:i] + report[i + 1 :]
+            if is_safe(dampened_report):
+                return True
+        return False
+
+    return sum(is_dampened_safe(report) for report in parse(input))
+
+
+def main() -> None:
+    input = sys.stdin.read().splitlines()
+    print(solve(input))
+
+
+if __name__ == "__main__":
+    main()