Day 07

day07/part2.py

@@ -0,0 +1,8 @@
+#!/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)))