42 lines
1.0 KiB
Python
42 lines
1.0 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
from sys import argv
|
||
|
|
from collections import Counter, deque
|
||
|
|
|
||
|
|
def attempt_one(line, iterations):
|
||
|
|
fishes = [int(fish) for fish in line if fish.isdigit()]
|
||
|
|
|
||
|
|
for days in range(iterations):
|
||
|
|
number_zeroes = fishes.count(0)
|
||
|
|
fishes = [fish if fish else 7 for fish in fishes]
|
||
|
|
if number_zeroes:
|
||
|
|
fishes.extend(number_zeroes * [9])
|
||
|
|
fishes = [fish-1 for fish in fishes]
|
||
|
|
return len(fishes)
|
||
|
|
|
||
|
|
|
||
|
|
def lanternfish(line, iterations):
|
||
|
|
fishes = Counter([int(fish) for fish in line if fish.isdigit()])
|
||
|
|
fishes_list = [0] * 9
|
||
|
|
for fish in fishes:
|
||
|
|
fishes_list[fish] = fishes[fish]
|
||
|
|
|
||
|
|
fishes_que = deque(fishes_list)
|
||
|
|
|
||
|
|
for days in range(iterations):
|
||
|
|
number_zeroes = fishes_que.popleft()
|
||
|
|
fishes_que.append(number_zeroes)
|
||
|
|
fishes_que[6] += number_zeroes
|
||
|
|
return sum(fishes_que)
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
with open("day6", "r") as file:
|
||
|
|
line = file.readline()
|
||
|
|
|
||
|
|
print(lanternfish(line, 80))
|
||
|
|
print(lanternfish(line, 256))
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|