day 6: simplified

1 files changed, 5 insertions, 9 deletions

diff --git a/6.py b/6.py
index 4e1e127..b290710 100644
--- a/6.py
+++ b/6.py
@@ -4,13 +4,9 @@ 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)
+def count_fish(life):
+    if life < 0: return 1
+    return count_fish(life - 7) + count_fish(life - 9)
 
-print(sum(count_fish(f,  79) for f in fish))
-print(sum(count_fish(f, 255) for f in fish))
+print(sum(count_fish(79 - f) for f in fish))
+print(sum(count_fish(255 - f) for f in fish))