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))