From 68110bf10a1d66655ffad7c9f58d868aefa94407 Mon Sep 17 00:00:00 2001 From: mhsn Date: Tue, 3 Dec 2024 12:00:00 +0000 Subject: 2024-03 python p1,p2 --- 2024/03/python/main.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) create mode 100644 2024/03/python/main.py (limited to '2024/03/python') 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) -- cgit v1.2.3