From a24c5ae084fbc502f68413005562586ce5dbf8fa Mon Sep 17 00:00:00 2001
From: Thibaud <thibaud.gasser@gmx.com>
Date: Wed, 11 Dec 2024 00:38:50 +0100
Subject: [PATCH] 2024 day 7

---
 2024/day7/day7.py | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 2024/day7/day7.py

diff --git a/2024/day7/day7.py b/2024/day7/day7.py
new file mode 100644
index 0000000..4fc8b3e
--- /dev/null
+++ b/2024/day7/day7.py
@@ -0,0 +1,36 @@
+def test(values, part2=False):
+    if len(values) == 1:
+        yield values[0]
+    else:
+        for rest in test(values[1:], part2):
+            yield values[0] + rest
+            yield values[0] * rest
+            if part2:
+                yield int(str(rest) + str(values[0]))  # concatenation
+
+
+def main(data):
+    part1 = 0
+    part2 = 0
+    for expected, values in data:
+        for res in test(values[::-1]):
+            if res == expected:
+                part1 += res
+                break
+        for res in test(values[::-1], part2=True):
+            if res == expected:
+                part2 += res
+                break
+
+    print("Part 1: ", part1)
+    print("Part 2: ", part2)
+
+
+if __name__ == "__main__":
+    import sys
+    infile = sys.argv[1] if 1 < len(sys.argv) else "example.txt"
+    with open(infile) as f:
+        data = [l.rstrip().split(": ") for l in f.readlines()]
+        data = [(int(a), tuple(map(int, b.split(" ")))) for a, b in data]
+        main(data)
+