solve day 6 (wew)

kwout 2 years ago
parent 8800495b02
commit 5f75458893
  1. 29
      2021/day6/6.py
  2. 1
      2021/day6/input.txt.sample

@ -0,0 +1,29 @@
fish = [int(i) for i in open("input.txt", 'r').read().split(",")]
fishies = {}
newFishies = {}
for i in range(9):
fishies[i] = 0
newFishies[i] = 0
for f in fish:
fishies[f] += 1
for i in range(int(input("days: "))):
tempFish = fishies.copy()
fishies[6] = fishies[0]+newFishies[0]
for j in range(6):
fishies[j] = tempFish[j+1]
foo = tempFish[0]
tempFish = newFishies.copy()
newFishies[8] = foo + newFishies[0]
for j in range(8):
newFishies[j] = tempFish[j+1]
summa = 0
for x in fishies:
summa += fishies[x]
for x in newFishies:
summa += newFishies[x]
print(summa)
Loading…
Cancel
Save