54 lines
1.1 KiB
Python
54 lines
1.1 KiB
Python
|
import re
|
||
|
|
|
||
|
|
from output import answer # , matrix, D, DD, ADJ, ints, mhd, mdbg, vdbg
|
||
|
|
|
||
|
|
n = 15
|
||
|
|
title = "Timing is Everything"
|
||
|
|
|
||
|
|
|
||
|
|
@answer(1, "First time for capsule-giving button press is {}")
|
||
|
|
def part_1(presolved):
|
||
|
|
return presolved[0]
|
||
|
|
|
||
|
|
|
||
|
|
@answer(2, "With the additional disc, first time for capsule-giving button press is {}")
|
||
|
|
def part_2(presolved):
|
||
|
|
return presolved[1]
|
||
|
|
|
||
|
|
|
||
|
|
def solve(data):
|
||
|
|
M = [
|
||
|
|
tuple(map(int, re.search(r"(\d+) pos.+ion (\d+)", r).groups()))
|
||
|
|
for r in data.splitlines()
|
||
|
|
]
|
||
|
|
p1 = wait_and_press(M)
|
||
|
|
p2 = wait_and_press(M + [(11, 0)])
|
||
|
|
return p1, p2
|
||
|
|
|
||
|
|
|
||
|
|
def wait_and_press(M):
|
||
|
|
t = 0
|
||
|
|
while True:
|
||
|
|
if all((ti + lp[1]) % lp[0] == 0 for ti, lp in enumerate(M, start=t + 1)):
|
||
|
|
break
|
||
|
|
t += 1
|
||
|
|
return t
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
inp = """
|
||
|
|
Disc #1 has 5 positions; at time=0, it is at position 4.
|
||
|
|
Disc #2 has 2 positions; at time=0, it is at position 1.
|
||
|
|
""".strip()
|
||
|
|
|
||
|
|
with open("./input/15.txt", "r") as f:
|
||
|
|
inp = f.read().strip()
|
||
|
|
|
||
|
|
inp = solve(inp)
|
||
|
|
|
||
|
|
a = part_1(inp)
|
||
|
|
b = part_2(inp)
|
||
|
|
|
||
|
|
assert a == 122318
|
||
|
|
assert b == 3208583
|