mirror of
https://github.com/Crocmagnon/advent-of-code.git
synced 2024-11-05 14:23:58 +01:00
Solve day 13 part 1 and nearly 2
This commit is contained in:
parent
66e11e3dac
commit
0dcf0cb22c
8 changed files with 83 additions and 0 deletions
69
2020/day13_bus.py
Normal file
69
2020/day13_bus.py
Normal file
|
@ -0,0 +1,69 @@
|
|||
import enum
|
||||
import functools
|
||||
import math
|
||||
from typing import List, Dict
|
||||
|
||||
|
||||
def main(filename: str, expected_part_1: int = None, expected_part_2: int = None):
|
||||
print(f"\n+ Running on {filename}")
|
||||
with open(filename) as f:
|
||||
notes = f.read().strip().split("\n")
|
||||
|
||||
counter_part_1 = solve_part_1(notes)
|
||||
|
||||
print(f"1. Found {counter_part_1}")
|
||||
if expected_part_1:
|
||||
assert expected_part_1 == counter_part_1
|
||||
|
||||
counter_part_2 = solve_part_2(notes)
|
||||
print(f"2. Found {counter_part_2}")
|
||||
if expected_part_2:
|
||||
assert expected_part_2 == counter_part_2
|
||||
|
||||
|
||||
def solve_part_1(notes) -> int:
|
||||
earliest = int(notes[0])
|
||||
buses = map(int, filter(lambda bus: bus != "x", notes[1].split(",")))
|
||||
min_delta = math.inf
|
||||
min_bus = None
|
||||
for bus in buses:
|
||||
delta = math.ceil(earliest / bus) * bus - earliest
|
||||
if delta < min_delta:
|
||||
min_delta = delta
|
||||
min_bus = bus
|
||||
|
||||
return min_bus * min_delta
|
||||
|
||||
|
||||
def solve_part_2(notes) -> int:
|
||||
"""Works but far too slow for the real deal."""
|
||||
split = notes[1].split(",")
|
||||
buses = list(map(int, filter(lambda bus: bus != "x", split)))
|
||||
minutes = {} # type: Dict[int, int]
|
||||
biggest_bus = max(buses)
|
||||
biggest_bus_index = split.index(str(biggest_bus))
|
||||
|
||||
for bus in buses:
|
||||
minutes[split.index(str(bus)) - biggest_bus_index] = bus
|
||||
|
||||
found = False
|
||||
timestamp = 99999999999628
|
||||
while not found:
|
||||
timestamp += biggest_bus
|
||||
found = True
|
||||
for delta, bus in minutes.items():
|
||||
if (timestamp + delta) % bus != 0:
|
||||
found = False
|
||||
break
|
||||
|
||||
return timestamp + min(minutes.keys())
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main("inputs/day13-test1", 295, 1068781)
|
||||
main("inputs/day13-test2", None, 3417)
|
||||
main("inputs/day13-test3", None, 754018)
|
||||
main("inputs/day13-test4", None, 779210)
|
||||
main("inputs/day13-test5", None, 1261476)
|
||||
main("inputs/day13-test6", None, 1202161486)
|
||||
main("inputs/day13", 4315)
|
2
2020/inputs/day13
Normal file
2
2020/inputs/day13
Normal file
|
@ -0,0 +1,2 @@
|
|||
1001938
|
||||
41,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,37,x,x,x,x,x,431,x,x,x,x,x,x,x,23,x,x,x,x,13,x,x,x,17,x,19,x,x,x,x,x,x,x,x,x,x,x,863,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,x,29
|
2
2020/inputs/day13-test1
Normal file
2
2020/inputs/day13-test1
Normal file
|
@ -0,0 +1,2 @@
|
|||
939
|
||||
7,13,x,x,59,x,31,19
|
2
2020/inputs/day13-test2
Normal file
2
2020/inputs/day13-test2
Normal file
|
@ -0,0 +1,2 @@
|
|||
939
|
||||
17,x,13,19
|
2
2020/inputs/day13-test3
Normal file
2
2020/inputs/day13-test3
Normal file
|
@ -0,0 +1,2 @@
|
|||
939
|
||||
67,7,59,61
|
2
2020/inputs/day13-test4
Normal file
2
2020/inputs/day13-test4
Normal file
|
@ -0,0 +1,2 @@
|
|||
939
|
||||
67,x,7,59,61
|
2
2020/inputs/day13-test5
Normal file
2
2020/inputs/day13-test5
Normal file
|
@ -0,0 +1,2 @@
|
|||
939
|
||||
67,7,x,59,61
|
2
2020/inputs/day13-test6
Normal file
2
2020/inputs/day13-test6
Normal file
|
@ -0,0 +1,2 @@
|
|||
939
|
||||
1789,37,47,1889
|
Loading…
Reference in a new issue