From 7f42d47cae61824b2e88baa8003336380f24f855 Mon Sep 17 00:00:00 2001
From: Xiretza <xiretza@xiretza.xyz>
Date: Mon, 6 Dec 2021 19:35:16 +0100
Subject: [PATCH] 2021 day6/python: add solution

---
 2021/day6/day6.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)
 create mode 100755 2021/day6/day6.py

diff --git a/2021/day6/day6.py b/2021/day6/day6.py
new file mode 100755
index 0000000..cd62018
--- /dev/null
+++ b/2021/day6/day6.py
@@ -0,0 +1,20 @@
+#!/usr/bin/env python
+
+import sys
+import functools
+
+@functools.cache
+def spawns(days):
+    acc = days//7
+    days -= 9
+    acc += sum(spawns(days-x*7) for x in range(days//7))
+    return acc
+
+if __name__ == '__main__':
+    inp = [int(i) for i in sys.stdin.readline().split(',')]
+
+    def do(days):
+        spawns.cache_clear()
+        return sum(1 + spawns(days + (6 - fish)) for fish in inp)
+    print(do(80))
+    print(do(256))