Day 9.

2021-12-09 16:53:29 -05:00
parent d4204d5c8d
commit a31f009abe
4 changed files with 244 additions and 0 deletions
--- a/2021/9/main.py
+++ b/2021/9/main.py
@@ -0,0 +1,44 @@
+data = []
+
+with open('data', 'r') as fp:
+    # NOTE(dev): We assume all lines are the same length
+    for line in fp:
+        data_line = []
+        for ch in line.strip():
+            data_line.append(int(ch))
+        data.append(data_line)
+
+total = 0
+for y in range(0, len(data)):
+    for x in range(0, len(data[y])):
+        sub_total = 0
+        all_four = 0
+        # above
+        if(y > 0):
+            if(data[y-1][x] > data[y][x]):
+                all_four += 1
+        else:
+            all_four += 1 # skip
+        # below
+        if(y < len(data) - 1):
+            if(data[y+1][x] > data[y][x]):
+                all_four += 1
+        else:
+            all_four += 1 #skip
+        # left
+        if(x > 0):
+            if(data[y][x-1] > data[y][x]):
+                all_four += 1
+        else:
+            all_four += 1 #skip
+        # right
+        if(x < len(data[y]) - 1):
+            if(data[y][x+1] > data[y][x]):
+                all_four += 1
+        else:
+            all_four += 1 #skip
+
+        if(all_four == 4):
+            total += data[y][x] + 1
+
+print(total)