From f1ac55966465be0d50f1a64b3f329a8055a1dd8f Mon Sep 17 00:00:00 2001 From: Julia Lange Date: Mon, 2 Dec 2024 21:14:13 -0800 Subject: [PATCH] 2024 Dec 03rd --- Python/2024/03/main.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) create mode 100644 Python/2024/03/main.py diff --git a/Python/2024/03/main.py b/Python/2024/03/main.py new file mode 100644 index 0000000..ee37d18 --- /dev/null +++ b/Python/2024/03/main.py @@ -0,0 +1,24 @@ +import re +input = open("input", 'r').read().strip() + +# d = [line for line in input] +# d = [[val for val in line.strip()] for line in input] + +def run(part2 = False): + sm = 0 + x = re.findall(r"(mul\(\d*,\d*\))|(do\(\))|(don't\(\))", input) + enable = True + for item in x: + if enable and item[0]: + a, b = item[0].split(",") + a = int(a[4:]) + b = int(b[:-1]) + sm += a * b + elif item[1]: + enable = True + elif part2: + enable = False + return sm + +print(run()) +print(run(True))