Spaces:
Running
Running
File size: 1,767 Bytes
a4da721 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
def solve():
with open("./input.txt", "r") as f:
grid = [line.strip() for line in f]
rows = len(grid)
cols = len(grid[0])
def check_pattern(r, c, pattern):
for dr in [-1, 0, 1]:
for dc in [-1, 0, 1]:
if dr == 0 and dc == 0:
continue
found = True
for i in range(len(pattern)):
nr, nc = r + dr * i, c + dc * i
if 0 <= nr < rows and 0 <= nc < cols and grid[nr][nc] == pattern[i]:
continue
else:
found = False
break
if found:
return True
return False
# Part 1
count1 = 0
for r in range(rows):
for c in range(cols):
if check_pattern(r, c, "XMAS") or check_pattern(r, c, "SAMX"):
count1 += 1
print(count1)
# Part 2
count2 = 0
for r in range(rows):
for c in range(cols):
if (
1 <= r < rows - 1 and 1 <= c < cols - 1 and
(grid[r-1][c-1] == 'M' or grid[r-1][c-1] == 'S') and
(grid[r-1][c+1] == 'M' or grid[r-1][c+1] == 'S') and
(grid[r+1][c-1] == 'M' or grid[r+1][c-1] == 'S') and
(grid[r+1][c+1] == 'M' or grid[r+1][c+1] == 'S') and
(grid[r-1][c] == 'A' or grid[r-1][c] == 'A') and
(grid[r][c-1] == 'A' or grid[r][c-1] == 'A') and
(grid[r][c+1] == 'A' or grid[r][c+1] == 'A') and
(grid[r+1][c] == 'A' or grid[r+1][c] == 'A') and
(grid[r][c] == 'S' or grid[r][c] == 'M')
):
count2 += 1
print(count2)
solve() |