summaryrefslogtreecommitdiffstats
path: root/6.py
diff options
context:
space:
mode:
Diffstat (limited to '6.py')
-rw-r--r--6.py14
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))