Day 07 (glamorized)

This commit is contained in:
Sebastian Seedorf
2021-12-07 10:47:52 +01:00
parent de09a3a78c
commit 530d103717
2 changed files with 31 additions and 2 deletions

View File

@@ -4,5 +4,6 @@ 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)))
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)))