You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
17 lines
486 B
17 lines
486 B
import numpy as np
|
|
|
|
file = "inputs/day6.input"
|
|
numbers = [int(x) for x in open(file, "rb").read().decode().split(",")]
|
|
unique, counts = np.unique(numbers, return_counts=True)
|
|
fishies = [counts[unique == x][0] if (x in unique) else 0 for x in range(9)]
|
|
days = 256
|
|
|
|
for day in range(days):
|
|
labour_fishies = fishies[0]
|
|
for idx in range(len(fishies) - 1):
|
|
fishies[idx] = fishies[idx + 1]
|
|
fishies[6] += labour_fishies
|
|
fishies[8] = labour_fishies
|
|
|
|
print(sum(fishies))
|