aoc/2023
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Simplify day 5 further

Browse source
This commit is contained in:
Sijmen 2023-12-06 01:42:20 +01:00
parent 5cde3ab3b9
commit 8433c88147
1 changed files with 15 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
day05.py
View file

@ -33,34 +33,24 @@ def transform_ranges(map, ranges):
yield from transform_ranges(map, [(src_end, end - src_end)]) yield from transform_ranges(map, [(src_end, end - src_end)])
def do_the_thing(ranges): def do_the_thing(maps, ranges):
soils = transform_ranges(seed_to_soil, ranges) for map in maps:
ferts = transform_ranges(soil_to_fert, soils) ranges = transform_ranges(map, ranges)
waters = transform_ranges(fert_to_water, ferts) return min(ranges)[0]
lights = transform_ranges(water_to_light, waters)
temps = transform_ranges(light_to_temp, lights)
humids = transform_ranges(temp_to_humid, temps)
locs = transform_ranges(humid_to_loc, humids)
(loc, _) = min(locs)
return loc
def main():
input = fileinput.input() input = fileinput.input()
seeds = [int(seed) for seed in next(input).split()[1:]] seeds = [int(seed) for seed in next(input).split()[1:]]
next(input) next(input)
next(input) next(input)
seed_to_soil = list(parse_map(input)) maps = [list(parse_map(input)) for _ in range(7)]
soil_to_fert = list(parse_map(input))
fert_to_water = list(parse_map(input))
water_to_light = list(parse_map(input))
light_to_temp = list(parse_map(input))
temp_to_humid = list(parse_map(input))
humid_to_loc = list(parse_map(input))
part1 = do_the_thing((seed, 1) for seed in seeds) part1 = do_the_thing(maps, ((seed, 1) for seed in seeds))
part2 = do_the_thing(zip(*(iter(seeds),) * 2)) part2 = do_the_thing(maps, zip(*[iter(seeds)] * 2))
print(part1, part2) print(part1, part2)
main()