diff --git a/2021/day6/6.py b/2021/day6/6.py new file mode 100644 index 0000000..72b8aa8 --- /dev/null +++ b/2021/day6/6.py @@ -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)