initial commit

2021/Day6/Day6.py

@@ -0,0 +1,27 @@
+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))