Day 09 (Part 3)

day09/part3.py

@@ -1,7 +1,5 @@
 #!/usr/bin/env python3
-import math
 from math import comb
-
 from tqdm import tqdm
 
 lines = (x.strip() for x in open("input.txt"))
@@ -13,7 +11,7 @@ for line in tqdm(lines, total=200):
     for _ in range(365*5):
         numbers.append(sum(factor*val for factor, val in zip(combinations, numbers)))
         combinations.append(-1)
-        combinations = [-combinations[0] if i == 0 else combinations[i-1]-combinations[i] for i in range(len(combinations))]
+        combinations = [(combinations[i-1] if i > 0 else 0)-combinations[i] for i in range(len(combinations))]
     result += numbers[-1]
 
 print(result)