day 6

2 files changed, 17 insertions, 0 deletions

diff --git a/6.in b/6.in
new file mode 100644
index 0000000..de918c7
--- /dev/null
+++ b/6.in
@@ -0,0 +1 @@
+1,1,3,5,1,1,1,4,1,5,1,1,1,1,1,1,1,3,1,1,1,1,2,5,1,1,1,1,1,2,1,4,1,4,1,1,1,1,1,3,1,1,5,1,1,1,4,1,1,1,4,1,1,3,5,1,1,1,1,4,1,5,4,1,1,2,3,2,1,1,1,1,1,1,1,1,1,1,1,1,1,5,1,1,1,1,1,1,1,1,2,2,1,1,1,1,1,5,1,1,1,3,4,1,1,1,1,3,1,1,1,1,1,4,1,1,3,1,1,3,1,1,1,1,1,3,1,5,2,3,1,2,3,1,1,2,1,2,4,5,1,5,1,4,1,1,1,1,2,1,5,1,1,1,1,1,5,1,1,3,1,1,1,1,1,1,4,1,2,1,1,1,1,1,1,1,1,1,1,1,1,1,3,2,1,1,1,1,2,2,1,2,1,1,1,5,5,1,1,1,1,1,1,1,1,1,1,1,1,2,2,1,1,4,2,1,4,1,1,1,1,1,1,1,2,1,2,1,1,1,1,1,1,1,1,1,1,1,1,1,2,2,1,5,1,1,1,1,1,1,1,1,3,1,1,3,3,1,1,1,3,5,1,1,4,1,1,1,1,1,4,1,1,3,1,1,1,1,1,1,1,1,2,1,5,1,1,1,1,1,1,1,1,1,1,4,1,1,1,1
diff --git a/6.py b/6.py
new file mode 100644
index 0000000..4e1e127
--- /dev/null
+++ b/6.py
@@ -0,0 +1,16 @@
+from functools import cache
+from utils import open_day
+
+fish = list(map(int, open_day(6).read().split(',')))
+
+@cache
+def count_fish(n, life):
+    if life < 0:
+        return 1
+    if n == 0:
+        return count_fish(0, life - 7) + count_fish(2, life - 7)
+    else:
+        return count_fish(0, life - n)
+
+print(sum(count_fish(f,  79) for f in fish))
+print(sum(count_fish(f, 255) for f in fish))