Files
python-aoc-2021/day07/part2.py
Sebastian Seedorf 530d103717 Day 07 (glamorized)
2021-12-07 10:47:52 +01:00

10 lines
295 B
Python

#!/usr/bin/env python3
lines = (list(map(int, x.strip().split(','))) for x in open("input.txt"))
for line in lines:
line.sort()
mean = round(sum(line) / len(line))
# fix of by one error
print(min(sum(abs(x - m)*(abs(x - m)+1)//2 for x in line) for m in range(mean-1, mean+2)))