diff options
| -rw-r--r-- | 6.py | 14 |
1 files changed, 5 insertions, 9 deletions
@@ -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)) |