2024 day 3 part 1 and 2

README.md

@@ -15,7 +15,7 @@
 - 2021 - ★★★★★ ★★★★★ ★★★★★ ★★★
 - 2022 - ★★★★★ ★★★★★ ★★★★★ ★☆
 - 2023 - ★★★★★ ★★★★★ ★★★★★ ★★★☆
-- 2024 - ★★
+- 2024 - ★★★
 
 ## How to use
 

src/year2024/day03a.py

@@ -0,0 +1,8 @@
+"""2024 - Day 3 Part 1: Mull It Over"""
+
+import re
+
+
+def solve(task: str) -> int:
+    result = re.findall(r"mul\((\d+),(\d+)\)", task)
+    return sum(int(a) * int(b) for a, b in result)

src/year2024/day03b.py

@@ -0,0 +1,24 @@
+"""2024 - Day 3 Part 2: Mull It Over"""
+
+import re
+
+
+def solve(task: str) -> int:
+    result = re.findall(r"mul\((\d+),(\d+)\)|(do\(\))|(don't\(\))", task)
+
+    ans = 0
+    skip = False
+
+    for a, b, do, dnt in result:
+        if dnt:
+            skip = True
+        elif do:
+            skip = False
+        elif skip:
+            continue
+        else:
+            a = int(a)
+            b = int(b)
+            ans += a * b
+
+    return ans

tests/src/year2024/test_day03a.py

@@ -0,0 +1,8 @@
+"""2024 - Day 3 Part 1: Mull It Over"""
+
+from src.year2024.day03a import solve
+
+
+def test_solve():
+    task = "xmul(2,4)%&mul[3,7]!@^do_not_mul(5,5)+mul(32,64]then(mul(11,8)mul(8,5))"
+    assert solve(task) == 161

tests/src/year2024/test_day03b.py

@@ -0,0 +1,8 @@
+"""2024 - Day 3 Part 2: Mull It Over"""
+
+from src.year2024.day03b import solve
+
+
+def test_solve():
+    task = "xmul(2,4)&mul[3,7]!^don't()_mul(5,5)+mul(32,64](mul(11,8)undo()?mul(8,5))"
+    assert solve(task) == 48