Add solution for day 7

AoC2021/code/AoC07.py

@@ -0,0 +1,27 @@
+#!/usr/bin/env python3
+
+import sys
+
+
+def fuel(crabs, x):
+    return sum(abs(c-x) for c in crabs)
+
+def fuel_incr(crabs, x):
+    return sum(sc[abs(c-x)] for c in crabs)
+
+if __name__ == '__main__':
+    lines = open(sys.argv[1]).read().strip('\n\n').split('\n') # or split \n\n or sth similar
+    nums = [int(i) for i in lines[0].split(",")]
+
+    # challenge 1
+    res1 = str(min(fuel(nums, i) for i in range(max(nums))))
+    print("challenge 1:" +"\n" +  res1 + "\n")
+
+    # challenge 2
+    sc = [0] * (max(nums)+1)
+    for i in range(max(nums)+1):
+        sc[i] = sc[i-1] + i
+
+    res2 = str(min(fuel_incr(nums, i) for i in range(max(nums))))
+    print("challenge 2:" +"\n" +  res2 + "\n")
+