From: Maximilian Friedersdorff <max@friedersdorff.com>
Date: Tue, 7 Dec 2021 21:10:11 +0000 (+0000)
Subject: Do day 7
X-Git-Url: https://git.friedersdorff.com/?a=commitdiff_plain;h=08fc3d5715cdfa0e23c4dd6079a06d2598718177;hp=081d73d2bb77101d614400ed76212a4933a64217;p=max%2Fadvent_of_code_2021.git

Do day 7
---

diff --git a/7_1.py b/7_1.py
new file mode 100644
index 0000000..0c4557e
--- /dev/null
+++ b/7_1.py
@@ -0,0 +1,13 @@
+"""Find the least fuel for the sub-crabs"""
+with open("7_input.txt") as f:
+    crab_positions = [int(x) for x in f.read().split(",")]
+
+fuel_costs = []
+for target_position in range(max(crab_positions) + 1):
+    fuel_costs.append(0)
+    for crab_position in crab_positions:
+        fuel_costs[-1] += abs(target_position - crab_position)
+
+min_fuel = min(fuel_costs)
+min_fuel_position = fuel_costs.index(min_fuel)
+print(min_fuel_position, min_fuel)
diff --git a/7_2.py b/7_2.py
new file mode 100644
index 0000000..95592f7
--- /dev/null
+++ b/7_2.py
@@ -0,0 +1,14 @@
+"""Find the least fuel for the sub-crabs with triangle fuel consumption"""
+with open("7_input.txt") as f:
+    crab_positions = [int(x) for x in f.read().split(",")]
+
+fuel_costs = []
+for target_position in range(max(crab_positions) + 1):
+    fuel_costs.append(0)
+    for crab_position in crab_positions:
+        delta = abs(target_position - crab_position)
+        fuel_costs[-1] += delta * (delta + 1)/2
+
+min_fuel = min(fuel_costs)
+min_fuel_position = fuel_costs.index(min_fuel)
+print(min_fuel_position, min_fuel)