summaryrefslogtreecommitdiff
path: root/2024/03/python/main.py
diff options
context:
space:
mode:
Diffstat (limited to '2024/03/python/main.py')
-rw-r--r--2024/03/python/main.py22
1 files changed, 22 insertions, 0 deletions
diff --git a/2024/03/python/main.py b/2024/03/python/main.py
new file mode 100644
index 0000000..3713d85
--- /dev/null
+++ b/2024/03/python/main.py
@@ -0,0 +1,22 @@
+import re
+import sys
+
+with open(sys.argv[1], "r") as f:
+ s = f.read()
+
+silver = sum(int(x) * int(y) for x, y in re.findall(r"mul\((\d{1,3}),(\d{1,3})\)", s))
+
+gold = 0
+active = True
+for g in re.findall(r"(?:mul\((\d{1,3}),(\d{1,3})\))|(don't\(\))|(do\(\))", s):
+ match g, active:
+ case (x, y, "", ""), True:
+ gold += int(x) * int(y)
+ case (_, _, "don't()", _), _:
+ active = False
+ case (_, _, _, "do()"), _:
+ active = True
+
+
+print("silver:", silver)
+print("gold:", gold)