2021 day6/python: add solution

This commit is contained in:
Xiretza 2021-12-06 19:35:16 +01:00
parent 75d4a76a50
commit 7f42d47cae
1 changed files with 20 additions and 0 deletions

20
2021/day6/day6.py Executable file
View File

@ -0,0 +1,20 @@
#!/usr/bin/env python
import sys
import functools
@functools.cache
def spawns(days):
acc = days//7
days -= 9
acc += sum(spawns(days-x*7) for x in range(days//7))
return acc
if __name__ == '__main__':
inp = [int(i) for i in sys.stdin.readline().split(',')]
def do(days):
spawns.cache_clear()
return sum(1 + spawns(days + (6 - fish)) for fish in inp)
print(do(80))
print(do(256))