diff --git a/2024-python/aoc.py b/2024-python/aoc.py index 5040efa..c84d089 100644 --- a/2024-python/aoc.py +++ b/2024-python/aoc.py @@ -34,7 +34,7 @@ from collections import deque, Counter from heapq import heappop, heappush from itertools import compress, combinations, chain -from output import answer # , matrix, D, DD, ADJ, ints, mhd, mdbg, vdbg +from output import matrix, D, DD, ADJ, ints, mhd, mdbg, vdbg def solve(data): diff --git a/2024-python/output/__init__.py b/2024-python/output/__init__.py index d1d71d4..7e7bab1 100644 --- a/2024-python/output/__init__.py +++ b/2024-python/output/__init__.py @@ -127,6 +127,7 @@ def dijkstras(grid, start, target): all nodes. """ import heapq + target = max(grid) seen = set() queue = [(start, 0)] diff --git a/2024-python/output/day_03.py b/2024-python/output/day_03.py new file mode 100644 index 0000000..ee20ca8 --- /dev/null +++ b/2024-python/output/day_03.py @@ -0,0 +1,26 @@ +import re +from math import prod + + +def solve(data): + needle = re.compile(r"mul\((\d{1,3}),(\d{1,3})\)") + + p1 = sum(prod(map(int, factors)) for factors in re.findall(needle, data)) + + p2 = sum( + sum(prod(map(int, factors)) for factors in re.findall(needle, chunk)) + for chunk in data.split("do") + if not chunk.startswith("n't") + ) + + return p1, p2 + + +if __name__ == "__main__": + with open("./input/03.txt", "r") as f: + inp = f.read().strip() + + p1, p2 = solve(inp) + + print(p1) + print(p2)