Day 20

2021/day20.py

@@ -0,0 +1,38 @@
+from operator import itemgetter as ig
+
+with open("input20.txt") as f:
+    lines = [line.rstrip() for line in f]
+iea = lines[0]
+data = {
+    (x, y): c == "#"
+    for x, line in enumerate(lines[2:])
+    for y, c in enumerate(line)
+}
+
+
+def near(x, y):
+    return ((a, b) for a in range(x - 1, x + 2) for b in range(y - 1, y + 2))
+
+
+def enhance(image, unkv):
+    next_image = {}
+    xmin, ymin = min(image, key=ig(0))[0] - 1, min(image, key=ig(1))[1] - 1
+    xmax, ymax = max(image, key=ig(0))[0] + 1, max(image, key=ig(1))[1] + 1
+    for x in range(xmin, xmax + 1):
+        for y in range(ymin, ymax + 1):
+            v = 0
+            for nnp in near(x, y):
+                v = (v << 1) | int(image.get(nnp, unkv))
+            next_image[(x, y)] = iea[v] == "#"
+    return next_image
+
+
+def solve(num_steps):
+    image = data
+    for i in range(num_steps):
+        image = enhance(image, i % 2 == 1)
+    print(sum(image.values()))
+
+
+solve(2)
+solve(50)