27 lines
670 B
Python
27 lines
670 B
Python
def calc(fish: {}, days):
|
|
for i in range(256):
|
|
toAdd = fish["0"]
|
|
for key in "012345678":
|
|
if key != "8":
|
|
fish[key] = fish[str(int(key) + 1)]
|
|
|
|
fish["6"] = int(fish["6"]) + toAdd
|
|
fish["8"] = toAdd
|
|
|
|
sum = 0
|
|
for key in fish:
|
|
sum += fish[key]
|
|
|
|
return sum
|
|
|
|
|
|
if __name__ == '__main__':
|
|
with open("input.txt", "r") as f:
|
|
raw = f.read().strip().split(",")
|
|
|
|
fish = {"0": 0, "1": raw.count("1"), "2": raw.count("2"), "3": raw.count("3"), "4": raw.count("4"),
|
|
"5": raw.count("5"),
|
|
"6": raw.count("6"), "7": 0, "8": 0
|
|
}
|
|
|
|
print(calc(fish, 256))
|