Files
aocpy/2021/d7.py
2024-11-21 08:43:15 -05:00

21 lines
373 B
Python

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)