32 lines
842 B
Python
32 lines
842 B
Python
import itertools as it
|
|
import sys
|
|
|
|
lines = sys.stdin.read().strip().splitlines()
|
|
n = len(lines)
|
|
|
|
answer_1 = sum(
|
|
line.count("XMAS") + line.count("SAMX")
|
|
for i in range(n)
|
|
for ri, rk, ro, ci, ck, cm in (
|
|
(1, 0, 0, 0, 1, n),
|
|
(0, 1, 0, 1, 0, n),
|
|
(0, 1, 0, 1, 1, n - i),
|
|
(0, -1, -1, 1, 1, n - i),
|
|
(1, 1, 0, 0, 1, n - i if i != 0 else 0),
|
|
(-1, -1, -1, 0, 1, n - i if i != 0 else 0),
|
|
)
|
|
if (
|
|
line := "".join(lines[ri * i + rk * k + ro][ci * i + ck * k] for k in range(cm))
|
|
)
|
|
)
|
|
|
|
answer_2 = sum(
|
|
lines[i][j] == "A"
|
|
and "".join(lines[i + di][j + dj] for di, dj in it.product((-1, 1), (-1, 1)))
|
|
in {"MSMS", "SSMM", "MMSS", "SMSM"}
|
|
for i, j in it.product(range(1, n - 1), range(1, n - 1))
|
|
)
|
|
|
|
print(f"answer 1 is {answer_1}")
|
|
print(f"answer 2 is {answer_2}")
|