Finish day 2

diff --git a/2_1.py b/2_1.py
new file mode 100644 (file)
index 0000000..4ad6f27
--- /dev/null
+++ b/2_1.py
@@ -0,0 +1,19 @@
+"""Calculate the final position of the submarine"""
+
+horizonal = 0
+depth = 0
+with open("2_1_input.txt") as f:
+    for line in f:
+        command, distance = line.split(" ")
+        distance = int(distance)
+
+        if command == "down":
+            depth += distance
+        elif command == "up":
+            depth -= distance
+        elif command == "forward":
+            horizonal += distance
+        else:
+            raise RuntimeError("you fucked it captain")
+
+print(horizonal, depth, horizonal * depth)

diff --git a/2_2.py b/2_2.py
new file mode 100644 (file)
index 0000000..9c3b6a1
--- /dev/null
+++ b/2_2.py
@@ -0,0 +1,21 @@
+"""Calculate the final position of the submarine"""
+
+aim = 0
+horizonal = 0
+depth = 0
+with open("2_1_input.txt") as f:
+    for line in f:
+        command, distance = line.split(" ")
+        distance = int(distance)
+
+        if command == "down":
+            aim += distance
+        elif command == "up":
+            aim -= distance
+        elif command == "forward":
+            horizonal += distance
+            depth += aim * distance
+        else:
+            raise RuntimeError("you fucked it captain")
+
+print(horizonal, depth, horizonal * depth)