From 7d710ad3f92e8e85645b7a68b353d2b853bf111e Mon Sep 17 00:00:00 2001 From: Tomasz Kramkowski Date: Mon, 6 Dec 2021 05:37:17 +0000 Subject: day 6: simplified --- 6.py | 14 +++++--------- 1 file 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)) -- cgit v1.2.3-54-g00ecf