adventofcode-2021/06/solve02.py
2021-12-06 06:58:16 +01:00

19 lines
411 B
Python

#!/usr/bin/python
round=256
data = None
lanternfish = dict()
with open("input01.txt","r") as f:
data = [int(i) for i in f.readline().split(',')]
for i in range(10):
lanternfish[i] = data.count(i)
for r in range(round):
lanternfish[7] += lanternfish[0]
lanternfish[9] = lanternfish[0]
lanternfish = { k-1:v for (k,v) in lanternfish.items() if k > 0}
print(sum([v for (k,v) in lanternfish.items()]))