diff options
Diffstat (limited to '2024/03/python/main.py')
-rw-r--r-- | 2024/03/python/main.py | 22 |
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) |