Pascal Lais
71ebdf8f65
All checks were successful
continuous-integration/drone/push Build is passing
60 lines
1.5 KiB
Python
60 lines
1.5 KiB
Python
#!/usr/bin/env python3
|
|
from pathlib import Path
|
|
|
|
|
|
def enhance(algorithm, in_image, b):
|
|
out_image = []
|
|
for i in range(-b, len(in_image) + b):
|
|
line = ''
|
|
for j in range(-b, len(in_image[0]) + b):
|
|
idx = ''
|
|
for k in range(i - 1, i + 2):
|
|
for l in range(j - 1, j + 2):
|
|
if 0 <= k < len(in_image) and \
|
|
0 <= l < len(in_image[k]) and\
|
|
'#' == in_image[k][l]:
|
|
idx += '1'
|
|
else:
|
|
idx += '0'
|
|
|
|
idx = int(idx, 2)
|
|
line += algorithm[idx]
|
|
out_image.append(line)
|
|
return out_image
|
|
|
|
|
|
def part_1(input):
|
|
result = 0
|
|
algorithm = input[0].rstrip()
|
|
image = []
|
|
for line in input[2:]:
|
|
image.append(line.rstrip())
|
|
image = enhance(algorithm, image, 3)
|
|
image = enhance(algorithm, image, -1)
|
|
for line in image:
|
|
result += line.count('#')
|
|
|
|
print("Part 1 result:", result)
|
|
|
|
|
|
def part_2(input):
|
|
result = 0
|
|
algorithm = input[0].rstrip()
|
|
image = []
|
|
for line in input[2:]:
|
|
image.append(line.rstrip())
|
|
for _ in range(25):
|
|
image = enhance(algorithm, image, 3)
|
|
image = enhance(algorithm, image, -1)
|
|
for line in image:
|
|
result += line.count('#')
|
|
print("Part 2 result:", result)
|
|
|
|
|
|
input = list()
|
|
p = Path(__file__).with_name('input.txt')
|
|
with open(p) as f:
|
|
input = f.readlines()
|
|
part_1(input)
|
|
part_2(input)
|