generated from public/aoc_template
116 lines
3.6 KiB
Python
116 lines
3.6 KiB
Python
from tools.aoc import AOCDay
|
|
from typing import Any
|
|
|
|
|
|
def lookup(the_map: dict, source: int) -> int:
|
|
for (s, l), d in the_map.items():
|
|
if s <= source < s + l:
|
|
return source - s + d
|
|
return source
|
|
|
|
|
|
class Day(AOCDay):
|
|
inputs = [
|
|
[
|
|
(35, "input5_test"),
|
|
(346433842, "input5"),
|
|
],
|
|
[
|
|
(46, "input5_test"),
|
|
(60294664, "input5"),
|
|
],
|
|
]
|
|
|
|
def __init__(self, year: int, day: int):
|
|
super().__init__(year, day)
|
|
self.seed_locations = {}
|
|
|
|
def get_seed_location(self, seed: int) -> int:
|
|
soil = lookup(self.seed2soil, seed)
|
|
fertilizer = lookup(self.soil2fertilizer, soil)
|
|
water = lookup(self.fertilizer2water, fertilizer)
|
|
light = lookup(self.water2light, water)
|
|
temperature = lookup(self.light2temperature, light)
|
|
humidity = lookup(self.temperature2humidity, temperature)
|
|
location = lookup(self.humidity2location, humidity)
|
|
return location
|
|
|
|
def parse_input(self, part2: bool = False) -> list:
|
|
seeds = []
|
|
self.seed2soil = {}
|
|
self.soil2fertilizer = {}
|
|
self.fertilizer2water = {}
|
|
self.water2light = {}
|
|
self.light2temperature = {}
|
|
self.temperature2humidity = {}
|
|
self.humidity2location = {}
|
|
the_map = None
|
|
for line in self.getInput():
|
|
if not line:
|
|
continue
|
|
elif line.startswith("seeds:"):
|
|
_, s = line.split(": ")
|
|
seeds = list(map(int, s.split()))
|
|
elif line.startswith("seed-to-soil"):
|
|
the_map = self.seed2soil
|
|
elif line.startswith("soil-to-fertilizer"):
|
|
the_map = self.soil2fertilizer
|
|
elif line.startswith("fertilizer-to-water"):
|
|
the_map = self.fertilizer2water
|
|
elif line.startswith("water-to-light"):
|
|
the_map = self.water2light
|
|
elif line.startswith("light-to-temperature"):
|
|
the_map = self.light2temperature
|
|
elif line.startswith("temperature-to-humidity"):
|
|
the_map = self.temperature2humidity
|
|
elif line.startswith("humidity-to-location"):
|
|
the_map = self.humidity2location
|
|
else:
|
|
dest, source, length = map(int, line.split())
|
|
the_map[(source, length)] = dest
|
|
|
|
self.seed_locations = {}
|
|
last_min = 0
|
|
for (source, length), dest in sorted(self.seed2soil.items()):
|
|
if last_min < source:
|
|
self.seed2soil[(last_min, source - last_min)] = 0
|
|
last_min = source + length
|
|
self.seed2soil[(last_min, 2**32)] = 0
|
|
|
|
return seeds
|
|
|
|
def part1(self) -> Any:
|
|
seeds = self.parse_input()
|
|
return min(self.get_seed_location(s) for s in seeds)
|
|
|
|
def part2(self) -> Any:
|
|
return
|
|
seeds = self.parse_input()
|
|
min_loc = 2**32
|
|
for i in range(0, len(seeds), 2):
|
|
for s in range(seeds[i], seeds[i] + seeds[i + 1]):
|
|
x = self.get_seed_location(s)
|
|
if x < min_loc:
|
|
min_loc = x
|
|
print(i, "=>", min_loc)
|
|
|
|
return min_loc
|
|
|
|
|
|
if __name__ == "__main__":
|
|
day = Day(2023, 5)
|
|
day.run(verbose=True)
|
|
inst, nodes = self.parse_input()
|
|
cur_node = [x for x in nodes if x.endswith("A")]
|
|
cur_step = 0
|
|
while set(x[2] for x in cur_node) != {"Z"}:
|
|
new_node = []
|
|
dir = inst[cur_step % len(inst)]
|
|
for node in cur_node:
|
|
if dir == "L":
|
|
new_node.append(nodes[node][0])
|
|
else:
|
|
new_node.append(nodes[node][1])
|
|
|
|
cur_step += 1
|