blob: 9edf0823d524c0a725940edc4be9cc54369ce39c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
# pyright: strict
from sys import stdin
def predict_next(history: list[int]) -> int:
if all(e == 0 for e in history):
return 0
return history[-1] + predict_next(
[a - b for a, b in zip(history[1:], history[:-1])]
)
histories = [[int(e) for e in l.rstrip().split()] for l in stdin]
print(sum(predict_next(history) for history in histories))
print(sum(predict_next(history[::-1]) for history in histories))
|
