mikael.capelle/advent-of-code
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refactor 2024 day 3.

Browse Source
This commit is contained in:
Mikael CAPELLE 2024-12-03 15:22:37 +01:00
parent 0d50b44c37
commit c19279fad3
1 changed files with 7 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/holt59/aoc/2024/day3.py
View File

@ -4,19 +4,19 @@ from typing import Iterator
def extract_multiply(line: str) -> Iterator[int]: def extract_multiply(line: str) -> Iterator[int]:
for m in re.findall(r"mul\(([0-9]{1,3}),\s*([0-9]{1,3})\)", line): for m in re.finditer(r"mul\(([0-9]{1,3}),\s*([0-9]{1,3})\)", line):
yield int(m[0]) * int(m[1]) yield int(m.group(1)) * int(m.group(2))
def remove_disabled_parts(line: str) -> str: def valid_memory_blocks(line: str) -> Iterator[str]:
fixed_line, accumulate = "", True accumulate = True
while line: while line:
if accumulate: if accumulate:
if (dont_i := line.find("don't()")) != -1: if (dont_i := line.find("don't()")) != -1:
fixed_line += line[:dont_i] yield line[:dont_i]
line, accumulate = line[dont_i:], False line, accumulate = line[dont_i:], False
else: else:
fixed_line += line yield line
line = "" line = ""
else: else:
if (do_i := line.find("do()")) != -1: if (do_i := line.find("do()")) != -1:
@ -24,13 +24,11 @@ def remove_disabled_parts(line: str) -> str:
else: else:
line = "" line = ""
return fixed_line
line = sys.stdin.read().strip() line = sys.stdin.read().strip()
answer_1 = sum(extract_multiply(line)) answer_1 = sum(extract_multiply(line))
answer_2 = sum(extract_multiply(remove_disabled_parts(line))) answer_2 = sum(sum(extract_multiply(block)) for block in valid_memory_blocks(line))
print(f"answer 1 is {answer_1}") print(f"answer 1 is {answer_1}")
print(f"answer 2 is {answer_2}") print(f"answer 2 is {answer_2}")