1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
# Day Six: Lanternfish
file = open("input/06.txt").readlines()
ages = []
for num in file[0].split(","):
ages.append(int(num))
for day in range(80):
for i, fish in enumerate(ages):
if fish == 0:
ages[i] = 6
ages.append(9)
else:
ages[i] = fish-1
print(len(ages))
# ages: 0, 1, 2, 3, 4, 5, 6, 7, 8, 9
ages = [0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
for num in file[0].split(","):
ages[int(num)] += 1
for day in range(256):
for i, age in enumerate(ages):
if i == 0:
ages[7] += ages[0]
ages[9] += ages[0]
ages[0] -= ages[0]
else:
ages[i-1] += ages[i]
ages[i] -= ages[i]
print(sum(ages))
|
