2018 Day 11

Author: Programator2

2018/11a.py

@@ -0,0 +1,37 @@
+SERIAL = 7511
+# SERIAL = 57
+
+
+def compute_power_level(x, y):
+    id_ = x + 10
+    level = id_ * y + SERIAL
+    level *= id_
+    level = level % 1000 // 100
+    return level - 5
+
+
+def sum_3x3(x, y, matrix):
+    return sum(matrix[y_][x_] for x_ in range(x, x + 3) for y_ in range(y, y + 3))
+
+
+def main():
+    # print(compute_power_level(122,79))
+    # return
+    matrix = []
+    for y in range(1, 301):
+        matrix.append([compute_power_level(x, y) for x in range(1, 301)])
+    # print(matrix)
+    # return
+    print(
+        max(
+            (
+                (x, y, sum_3x3(x, y, matrix))
+                for x in range(0, 298)
+                for y in range(0, 298)
+            ),
+            key=lambda x: x[2],
+        )[:2]
+    )
+
+
+main()

2018/11b.py

@@ -0,0 +1,44 @@
+SERIAL = 7511
+
+
+def compute_power_level(x, y):
+    id_ = x + 10
+    level = id_ * y + SERIAL
+    level *= id_
+    level = level % 1000 // 100
+    return level - 5
+
+
+def sum_any(x, y, matrix, size):
+    return sum(matrix[y_][x_] for x_ in range(x, x + size) for y_ in range(y, y + size))
+
+
+def main():
+    matrix = []
+    for y in range(1, 301):
+        matrix.append([compute_power_level(x, y) for x in range(1, 301)])
+
+    remember = {(x, y, 1): matrix[y][x] for x in range(300) for y in range(300)}
+
+    max_entry, max_size = max(remember.items(), key=lambda x: x[1])
+
+    for x in range(300):
+        for y in range(300):
+            for size in range(2, 301):
+                if x + size > 300 or y + size > 300:
+                    continue
+                entry = (
+                    remember[(x, y, size - 1)]
+                    + sum(matrix[y+size-1][x_] for x_ in range(x, x + size))
+                    + sum(matrix[y_][x+size-1] for y_ in range(y, y + size - 1))
+                )
+                remember[(x, y, size)] = entry
+
+                if entry > max_size:
+                    max_size = entry
+                    max_entry = (x, y, size)
+        print(f'{x}')
+    print(max_entry)
+
+
+main()