Files
Advent2023/9/advent_1.py
Lauri Jesmin f5c4eacfb6 syntax fix
2024-08-08 14:49:21 +03:00

30 lines
561 B
Python
Executable File

#!/usr/bin/env python3.12
import re
def predict(numbers):
print(numbers)
result = []
for i in range(0, len(numbers)-1):
result.append(numbers[i+1]-numbers[i])
if len(set(result)) == 1:
print("ainult 1 tulemus")
return result[0] + numbers[-1]
return predict(result) + numbers[-1]
with open("data.txt") as file:
lines = [line.rstrip() for line in file]
y = 0
for line in lines:
temp = re.findall(r'\S+', line)
numbers = list(map(int, temp))
x = predict(numbers)
print(x)
y += x
print(y)