solution for day 7

1 files changed, 21 insertions, 0 deletions

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)