41 lines
962 B
Python
41 lines
962 B
Python
|
from output import answer
|
||
|
|
|
||
|
|
n = 9
|
||
|
|
title = "Mirage Maintenance"
|
||
|
|
|
||
|
|
|
||
|
|
@answer(1, "OASIS report extrapolated values sum is {}")
|
||
|
|
def part_1(data):
|
||
|
|
lines = [[int(d) for d in line.split()] for line in data.splitlines()]
|
||
|
|
return _solve(lines)
|
||
|
|
|
||
|
|
|
||
|
|
@answer(2, "Using prepending, OASIS report extrapolated values sum is {}")
|
||
|
|
def part_2(data):
|
||
|
|
lines = [[int(d) for d in line.split()[::-1]] for line in data.splitlines()]
|
||
|
|
return _solve(lines)
|
||
|
|
|
||
|
|
|
||
|
|
def _solve(lines):
|
||
|
|
vs = 0
|
||
|
|
for l in lines:
|
||
|
|
h = [l]
|
||
|
|
while any(n != 0 for n in h[-1]):
|
||
|
|
h.append([b - a for a, b in zip(h[-1], h[-1][1:])])
|
||
|
|
h = h[::-1]
|
||
|
|
for i in range(len(h)):
|
||
|
|
h[i].append(0 if i == 0 else h[i - 1][-1] + h[i][-1])
|
||
|
|
vs += h[-1][-1]
|
||
|
|
return vs
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
with open("./input/09.txt", "r") as f:
|
||
|
|
inp = f.read().strip()
|
||
|
|
|
||
|
|
a = part_1(inp)
|
||
|
|
b = part_2(inp)
|
||
|
|
|
||
|
|
assert a == 1702218515
|
||
|
|
assert b == 925
|