day 3 part 1

2018/day-3/python/main.py

@@ -0,0 +1,35 @@
+import os, sys, pyperclip, re
+
+# returns x, y, width, height
+def parse(string):
+    match = re.search(pattern, string)
+    # return int(match[1]), int(match[2]), int(match[3]), int(match[4])
+    return list(map(int, [match[i] for i in range(1, 5)]))
+
+# I only later figured out that the dimensions were simply 1000 x 1000
+# but this now allows it to be of any dimension
+def dimensions(data):
+    def bottomright(x, y, width, height):
+        return (x + width, height + y)    
+    bottomrights = list(map(lambda dat : bottomright(*dat), data))
+    matrix_x = max(bottomrights, key=lambda item : item[0])[0]
+    matrix_y = max(bottomrights, key=lambda item : item[1])[1]
+    return matrix_x, matrix_y
+
+path = os.path.join(sys.path[0], '..', 'input')
+data = open(path, 'r').read().split('\n')
+pattern = r'#\d+ @ (\d+),(\d+): (\d+)x(\d+)'
+data = list(map(parse, data))
+WIDTH, HEIGHT = dimensions(data)
+matrix = [[0 for y in range(WIDTH)] for x in range(HEIGHT)]
+
+for offset_x, offset_y, width, height in data:
+    for x in range(width):
+        for y in range(height):
+            # matrix[offset_x + x][offset_y + y] = min(2, 1 + matrix[offset_x + x][offset_y + y])
+            matrix[offset_x + x][offset_y + y] += 1
+            
+# count = sum(map(lambda x : x.count(2), matrix))
+count = sum(map(lambda item : len(list(filter(lambda i : i >= 2, item))), matrix))
+print(count)
+pyperclip.copy(count)