No need for defs

2024/03/solution.py

@@ -3,28 +3,20 @@ import re
 
 data = get_data(day=3, year=2024)
 
-def part_one():
-	regex = r"mul\((\d+),(\d+)\)"
-	matches = re.findall(regex, data, re.MULTILINE)
-
-	total = 0
-	for match in matches:
+# Part one
+matches = re.findall(r"mul\((\d+),(\d+)\)", data, re.MULTILINE)
+total = 0
+for match in matches:
 	total += int(match[0]) * int(match[1])
-	print(total)
+print(total)
 
-def part_two():
-	mul_regex = r"mul\((\d+),(\d+)\)"
-	do_str = "do()"
-	dont_str = "don't()"
-
-	matches = re.finditer(mul_regex, data, re.MULTILINE)
-
-	total = 0
-	for match in matches:
+# Part two
+matches = re.finditer(r"mul\((\d+),(\d+)\)", data, re.MULTILINE)
+total = 0
+for match in matches:
 	start = match.start()
-		do_match = data.rfind(do_str, 0, start)
-		dont_match = data.rfind(dont_str, 0, start)
-
+	do_match = data.rfind("do()", 0, start)
+	dont_match = data.rfind("don't()", 0, start)
 	if do_match == dont_match == -1: # Neither
 		total += int(match.group(1)) * int(match.group(2))
 		continue
@@ -33,8 +25,4 @@ def part_two():
 		continue
 	else: # Don't
 		continue
-
-	print(total)
-
-part_one()
-part_two()
+print(total)