2023: d09: ex1: add solution
This commit is contained in:
parent
6f31630af7
commit
d5842f92be
31
2023/d09/ex1/ex1.py
Executable file
31
2023/d09/ex1/ex1.py
Executable file
|
@ -0,0 +1,31 @@
|
||||||
|
#!/usr/bin/env python
|
||||||
|
|
||||||
|
import itertools
|
||||||
|
import sys
|
||||||
|
|
||||||
|
|
||||||
|
def solve(input: list[str]) -> int:
|
||||||
|
def parse_line(line: str) -> list[int]:
|
||||||
|
return [int(n) for n in line.split()]
|
||||||
|
|
||||||
|
def parse(input: list[str]) -> list[list[int]]:
|
||||||
|
return [parse_line(line) for line in input]
|
||||||
|
|
||||||
|
def extrapolate(sequence: list[int]) -> int:
|
||||||
|
diffs = [n - p for p, n in itertools.pairwise(sequence)]
|
||||||
|
if all(n == 0 for n in diffs):
|
||||||
|
return sequence[-1]
|
||||||
|
return sequence[-1] + extrapolate(diffs)
|
||||||
|
|
||||||
|
sequences = parse(input)
|
||||||
|
|
||||||
|
return sum(extrapolate(seq) for seq in sequences)
|
||||||
|
|
||||||
|
|
||||||
|
def main() -> None:
|
||||||
|
input = sys.stdin.read().splitlines()
|
||||||
|
print(solve(input))
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
Loading…
Reference in a new issue