advent2021/seven.py

13 lines
340 B
Python
Executable File

#!/usr/bin/env python3
from useful import *
def frob(s):
return abs(s) * (abs(s) + 1) // 2
with open(0) as f:
pos = [int(x) for x in f.read().strip().split(',')]
a, b = min(pos), max(pos)
print(min(sum(abs(n - i) for n in pos) for i in range(a, b+1)))
print(min(sum(frob(n - i) for n in pos) for i in range(a, b+1)))