39 lines
949 B
Python
39 lines
949 B
Python
|
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)
|