mirror of
https://github.com/thib8956/advent-of-code.git
synced 2024-12-26 05:46:29 +00:00
34 lines
896 B
Python
34 lines
896 B
Python
#!/usr/bin/env python3
|
|
from rules import part1_rules, part2_rules
|
|
|
|
|
|
def main(grid, rules):
|
|
generation = 0
|
|
while True:
|
|
changes, next_grid = step(grid, rules)
|
|
generation += 1
|
|
grid = next_grid
|
|
assert generation < 1000
|
|
if generation % 10 == 0:
|
|
print(f"Generation {generation}, changes: {changes}")
|
|
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=part2_rules))
|