advent2023/sixx.py

12 lines
308 B
Python
Executable File

#!/usr/bin/env python3
from useful import *
ts, ds = [numbers(l) for l in lines(open(0))]
def wins(t, d):
return t + 1 - 2 * next(dropwhile(lambda i: i*(t-i) <= d, range(t)))
print(np.prod([wins(t, d) for (t, d) in zip(ts, ds)]))
print(wins(*[int(''.join(str(el) for el in ar)) for ar in (ts, ds)]))