aoc-2021/6/6.py

19 lines
351 B
Python
Raw Normal View History

day_fish = [0] * 9
with open("input.txt") as f:
contents = f.read()
for fish in contents.split(","):
fish = int(fish)
day_fish[fish] += 1
print(day_fish)
days = 256
for i in range(0, days):
zeros = day_fish[0]
day_fish = day_fish[1:] + [0]
day_fish[6] += zeros
day_fish[8] += zeros
print(sum(day_fish))