From d7616fc88e78a9a1d6f7f2b912bd863e422e18e4 Mon Sep 17 00:00:00 2001 From: Tomasz Kramkowski Date: Tue, 7 Dec 2021 07:25:46 +0000 Subject: day 7: python --- 7.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) create mode 100644 7.py diff --git a/7.py b/7.py new file mode 100644 index 0000000..9457068 --- /dev/null +++ b/7.py @@ -0,0 +1,12 @@ +from utils import open_day +from statistics import median + +crabs = list(map(int, open_day(7).read().split(','))) + +p = int(median(sorted(crabs))) +print(sum(abs(n - p) for n in crabs)) +s = sum(crabs) +l = len(crabs) +t = (s + sum(crab * l > s for crab in crabs)) // l +def sigma_n(n): return n * (n + 1) // 2 +print(sum(sigma_n(abs(n - t)) for n in crabs)) -- cgit v1.2.3-54-g00ecf