simplify python file structure

1 files changed, 24 insertions, 0 deletions

diff --git a/2024/02/python.py b/2024/02/python.py
new file mode 100755
index 0000000..f2c807d
--- /dev/null
+++ b/2024/02/python.py
@@ -0,0 +1,24 @@
+#!/usr/bin/env python3
+
+from fileinput import input
+
+reports = [[int(level) for level in report.split()] for report in input()]
+
+
+def incr(xs):
+    return all(x < y and y - x <= 3 for x, y in zip(xs, xs[1:]))
+
+
+def safe(xs):
+    return incr(xs) or incr(xs[::-1])
+
+
+def drops(xs):
+    return (xs[:idx] + xs[idx + 1 :] for idx, _ in enumerate(xs))
+
+
+silver = sum(safe(rep) for rep in reports)
+gold = sum(any(safe(mod) for mod in drops(rep)) for rep in reports)
+
+print("silver:", silver)
+print("gold:", gold)