advent2022/eleven.py

29 lines
756 B
Python
Executable File

#!/usr/bin/env python3
from useful import *
def monkey(p):
ls = p.strip().split('\n')
op = ls[2].strip().split()[4]
return [op, *[numbers(l) for l in ls[1:3]], *[numbers(l)[0] for l in ls[3:]], 0]
ms = [monkey(p) for p in pgphs(open(0).read())]
def op(base, leop, vals):
if not vals: # happens once in our input; is 'old * old'
return base * base
return base * vals[0] if leop == '*' else base + vals[0]
def rnd():
for m in ms:
a = m[1]
m[6] += len(a)
for i in a:
i = op(i, m[0], m[2]) // 3
ms[m[4 if not (i % m[3]) else 5]][1].append(i)
#ic(ms)
m[1] = []
for i in range(20):
rnd()
print(reduce(lambda a, b: a * b, sorted(m[6] for m in ms)[-2:]))