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))