Solve 2021 day 7-11

2021/d7.py

@@ -0,0 +1,20 @@
+from lib import get_data
+from lib import ints
+
+data = get_data(__file__).strip()
+xs = ints(data)
+
+
+def d(t, x):
+    n = abs(t - x)
+    return n * (n + 1) // 2
+
+
+s_min_1 = 10**16
+s_min_2 = 10**16
+for t in range(min(xs), max(xs) + 1):
+    s_min_1 = min(s_min_1, sum(abs(t - x) for x in xs))
+    s_min_2 = min(s_min_2, sum(d(t, x) for x in xs))
+
+print(s_min_1)
+print(s_min_2)