From 0cd5da6a091bb07434180da50db178e61c911eb4 Mon Sep 17 00:00:00 2001 From: Brian Chu Date: Mon, 6 Dec 2021 22:01:50 -0800 Subject: solution for day 7 --- day7.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 day7.py (limited to 'day7.py') diff --git a/day7.py b/day7.py new file mode 100644 index 0000000..1a2d206 --- /dev/null +++ b/day7.py @@ -0,0 +1,21 @@ +#!/usr/bin/env python + +import numpy as np + +def triangle(num): + return (num * (num + 1)) / 2 + +with open("day7.txt") as data: + distances = np.array([num for num in map(int, next(data).strip().split(','))]).astype(int) + # part 1 + median = np.median(distances) + print(np.sum(np.abs(distances - median).astype(int))) + + # part 2 + minim, maxim = np.amin(distances), np.amax(distances) + least_dist = None + for i in range(minim, maxim + 1): + dist = np.sum(triangle(np.abs(distances - i)).astype(int)) + if least_dist is None or least_dist > dist: + least_dist = dist + print(least_dist) -- cgit 1.4.1-2-gfad0