12 lines
332 B
Python
Executable File
12 lines
332 B
Python
Executable File
#!/usr/bin/env python3
|
|
from useful import *
|
|
|
|
with open(0) as f:
|
|
|
|
fishlist = [int(x) for x in f.read().strip().split(',')]
|
|
gen = deque(np.pad(np.bincount(fishlist), (0, 9))[:9])
|
|
for i in range(int(sys.argv[1]) if len(sys.argv) == 2 else 80):
|
|
gen.append(gen.popleft())
|
|
gen[6] += gen[8]
|
|
print(sum(gen))
|