Files
python-aoc-2021/day07/part2.py
Sebastian Seedorf de09a3a78c Day 07
2021-12-07 10:39:21 +01:00

9 lines
275 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()
median = round(sum(line) / len(line))
print(min(sum(abs(x - m)*(abs(x - m)+1)//2 for x in line) for m in range(median-1, median+2)))