Day 3

2020/03.py

@@ -0,0 +1,49 @@
+def tree_counter(horizontal_shift, vertical_shift, input_list):
+    tree_count = 0
+    row_count = 0
+    for i in range(0, len(input_list), vertical_shift):
+        horizontal = row_count * horizontal_shift
+        vertical = i
+
+        reference = horizontal % len(input_list[i])
+        if input_list[i][reference] == "#":
+            tree_count += 1
+
+        temp_input = input_list[i]
+        if temp_input[reference] == "#":
+            temp_input = temp_input[:reference] + "X" + temp_input[reference + 1:]
+        if temp_input[reference] == ".":
+            temp_input = temp_input[:reference] + "O" + temp_input[reference + 1:]
+        
+        row_count += 1
+        print(f'''"Row {f"""{vertical}/{len(input_list) - 1}""":8} {temp_input:35} {i:5} {horizontal:5} {vertical:5} {tree_count:4}''')
+    return tree_count
+
+# with open('./2020/03.test') as input:
+with open('./2020/03.input') as input:
+    input_list = input.read().split('\n')
+    
+tree_count = 0
+for i in range(len(input_list)):
+    horizontal = i * 3
+    vertical = i
+
+    reference = horizontal % len(input_list[i])
+    if input_list[i][reference] == "#":
+        tree_count += 1
+print(tree_count)
+
+slope_list = []
+slope_list.append(tree_counter(1, 1, input_list))
+slope_list.append(tree_counter(3, 1, input_list))
+slope_list.append(tree_counter(5, 1, input_list))
+slope_list.append(tree_counter(7, 1, input_list))
+slope_list.append(tree_counter(1, 2, input_list))
+
+total = slope_list[0]
+
+for i in range(1, len(slope_list)):
+    total *= slope_list[i]
+
+print(slope_list)
+print(total)