add solution for 2021/02

Author: SH1RL0CK

2020/test.py

@@ -1,4 +1,4 @@
-with open("01_puzzle_input.txt", "r") as file:
+with open("2021/01_puzzle_input.txt", "r") as file:
     content = file.readlines()
 
 increased = 0

2021/01_puzzle1.py

@@ -1,4 +1,4 @@
-with open("01_puzzle_input.txt", "r") as file:
+with open("2021/01_puzzle_input.txt", "r") as file:
     content = file.readlines()
 
 increased = 0

2021/01_puzzle2.py

@@ -0,0 +1,13 @@
+content = open("2021/02_puzzle_input.txt", "r").read().splitlines()
+
+depth = 0
+horizontal = 0
+
+for line in content:
+    if line.startswith("forward "):
+        horizontal += int(line.split(" ")[1])
+    elif line.startswith("down "):
+        depth += int(line.split(" ")[1])
+    elif line.startswith("up "):
+        depth -= int(line.split(" ")[1])
+print(horizontal * depth)

2021/02_puzzle1.py

@@ -0,0 +1,16 @@
+content = open("2021/02_puzzle_input.txt", "r").read().splitlines()
+
+depth = 0
+horizontal = 0
+aim = 0
+
+for line in content:
+    if line.startswith("forward "):
+        value = int(line.split(" ")[1])
+        horizontal += value
+        depth += aim * value
+    elif line.startswith("down "):
+        aim += int(line.split(" ")[1])
+    elif line.startswith("up "):
+        aim -= int(line.split(" ")[1])
+print(horizontal * depth)