fixed stoopid elves (pizza was great btw)

1 files changed, 5 insertions, 9 deletions

diff --git a/2023/21/solve.py b/2023/21/solve.py
index a10340e..8c87408 100644
--- a/2023/21/solve.py
+++ b/2023/21/solve.py
@@ -53,19 +53,15 @@ while q:
             pre[n] += 1
 
 print(part1)
-print(pre)
 
 poly = {}
-for i in range(PREHEAT):
-    j = C + i * W
-    if j % 2:
-        poly[i] = pre[j]
+for i, p in pre.items():
+    if (i + 1) % W == C:
+        poly[i + 1] = p
 
-print(poly)
 f = Polynomial.fit(list(poly.keys()), list(poly.values()), 2)
-x = (FIN - C) / W
-res = f(x)
-print(f, x, round(res))
+res = f(FIN)
+print(f, round(res))
 
 # for i in range(PREHEAT, (FIN - C) / W + 1):
 #     poly.append(predict(poly))