birdulon
/
AdventOfCode
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

2018 Day 17 not entirely garbage edition

This commit is contained in:
Luke Hubmayer-Werner 2018-12-17 20:26:47 +10:30
parent ba4065cc1a
commit 79befca869
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
2018/day17.py
View File

@ -27,17 +27,19 @@ for xmin, xmax, ymin, ymax in zip(Xmin, Xmax, Ymin, Ymax):
grid = init_grid.copy() grid = init_grid.copy()
drop_point_blacklist = set()
def drop_water(source=spring, skip_cliffs=set()): def drop_water(source=spring, skip_cliffs=set()):
# TODO: The exit condition on this is borked and it will probably run forever. Killing it after like a minute and then running the things at the end gives the right answers. if tuple(source) in drop_point_blacklist:
# I'll probably revisit this properly later return False
x, y = source x, y = source
# print('Dropping water from', source) # print('Dropping water from', source)
surface = (grid[x, y:] > 0).argmax() surface = (grid[x, y:] > 0).argmax()
if surface == 0: if surface == 0:
if (grid[x, y-1:] > 0).argmax() == 0: if (grid[x, y-1:] > 0).argmax() == 0:
grid[x, y:] = -1 grid[x, y:] = -1
drop_point_blacklist.add(tuple(source))
return False return False
# raise ValueError('Out of bounds') # raise ValueError('Out of bounds')
surface_y = surface + y surface_y = surface + y
@ -77,6 +79,10 @@ def drop_water(source=spring, skip_cliffs=set()):
if cliff and (cliff not in skip_cliffs): if cliff and (cliff not in skip_cliffs):
cliffs.add((x + cliff, surface_y)) cliffs.add((x + cliff, surface_y))
if cliffs and len([c for c in cliffs if tuple(c) in drop_point_blacklist]) == len(cliffs):
drop_point_blacklist.add(tuple(source))
return False
succeeded = False succeeded = False
while sum([drop_water(c) for c in cliffs]): while sum([drop_water(c) for c in cliffs]):
succeeded = True succeeded = True