Skip to main content

2021 - Day 6

🎄 Advent of Code in Zef 🎄

https://adventofcode.com/2021/day/6

from zef import *  
from zef.ops import *
c = collect

ages = [3,4,3,1,2]

def dict_step(d: dict) -> dict:
def num(n):
if n==8: return d.get(0,0)
elif n==6: return d.get(7,0) + d.get(0,0)
else: return d.get(n+1,0)
return {n: num(n) for n in range(9)}

result = (ages
| frequencies
| iterate[dict_step]
| nth[256]
| values
| sum
| collect
)