Solve day02 part 2

2021-12-04 15:47:33 -05:00 · 2021-12-04 15:47:33 -05:00 · 1e0e651fea
parent 9100b0f5d9
commit 1e0e651fea
2 changed files with 1017 additions and 0 deletions
--- a/day02.in2
+++ b/day02.in2
--- a/day02.py
+++ b/day02.py
@ -10,3 +10,20 @@ with open("day02.in") as f:

 print("Current position:", pos)
 print("Product:", pos[0] * pos[1])
+
+# Part 2
+pos = [0, 0, 0]
+with open("day02.in2") as f:
+    for x in f:
+        cmd = x.split()[0]
+        val = int(x.split()[1])
+        if cmd == "down":
+            pos[2] += val
+        elif cmd == "up":
+            pos[2] -= val
+        elif cmd == "forward":
+            pos[0] += val
+            pos[1] += pos[2] * val
+
+print("Current position:", pos)
+print("Product:", pos[0] * pos[1])