2024 d3: Add original and clean solution

Author: mebeim

2024/original_solutions/day03.py

@@ -0,0 +1,35 @@
+#!/usr/bin/env python3
+
+from utils.all import *
+
+advent.setup(2024, 3)
+
+fin = advent.get_input()
+data = fin.read()
+
+ans1 = ans2 = 0
+
+for a, b in re.findall(r'mul\((\d+),(\d+)\)', data):
+	a = int(a)
+	b = int(b)
+	ans1 += a * b
+
+advent.print_answer(1, ans1)
+
+enabled = True
+
+for m in re.findall(r'mul\((\d+),(\d+)\)|(don\'t\(\))|(do\(\))', data):
+	if m[2] == 'don\'t()':
+		enabled = False
+		continue
+	elif m[3] == 'do()':
+		enabled = True
+		continue
+
+	if enabled:
+		a, b = m[:2]
+		a = int(a)
+		b = int(b)
+		ans2 += a * b
+
+advent.print_answer(2, ans2)

2024/solutions/day03.py

@@ -0,0 +1,21 @@
+#!/usr/bin/env python3
+
+import sys
+from re import findall
+
+# Open the first argument as input or use stdin if no arguments were given
+fin = open(sys.argv[1]) if len(sys.argv) > 1 else sys.stdin
+
+total1 = total2 = 0
+enabled = True
+
+for a, b, do, dont in findall(r"mul\((\d+),(\d+)\)|(do\(\))|(don't\(\))", fin.read()):
+	if do or dont:
+		enabled = bool(do)
+	else:
+		x = int(a) * int(b)
+		total1 += x
+		total2 += x * enabled
+
+print('Part 1:', total1)
+print('Part 2:', total2)