156AC - ABC054B: Template Matching
ABC054-B (200 points)
問題
の行列 と の行列 が与えられたとき (), が に含まれるかどうかを判定する.
方針
- 総当たり.
import numpy as np # input N, M = map(int, input().split()) A = [list(input()) for _ in range(N)] B = [list(input()) for _ in range(M)] A_arr = np.array(A) B_arr = np.array(B) for i in range(N - M + 1): for j in range(N - M + 1): if np.array_equal(A_arr[i:i + M, j:j + M], B_arr): print('Yes') exit() print('No')