forked from thib8956/advent-of-code-2k20
33 lines
825 B
Python
33 lines
825 B
Python
|
#!/usr/bin/env python3
|
||
|
|
from rules import part1_rules
|
||
|
|
|
||
|
|
|
||
|
|
def main(grid, rules):
|
||
|
|
generation = 0
|
||
|
|
while True:
|
||
|
|
changes, next_grid = step(grid, rules)
|
||
|
|
generation += 1
|
||
|
|
grid = next_grid
|
||
|
|
assert generation < 1000
|
||
|
|
if changes == 0:
|
||
|
|
return next_grid.count("#")
|
||
|
|
|
||
|
|
|
||
|
|
def step(grid, rules):
|
||
|
|
changes = 0
|
||
|
|
next_grid = grid[:]
|
||
|
|
for index, cell in enumerate(grid):
|
||
|
|
try:
|
||
|
|
changes += rules[cell](index, grid, next_grid)
|
||
|
|
except KeyError:
|
||
|
|
pass
|
||
|
|
return changes, next_grid
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
with open("input.txt") as infile:
|
||
|
|
grid = list("".join(infile.read().splitlines()))
|
||
|
|
print("Part 1 ", main(grid, rules=part1_rules))
|
||
|
|
|
||
|
|
# print("Part 2 ", main(grid, rules={"L": handle_empty_2, "#": handle_occupied_2}))
|