[python]ロックとキー-2020ココア公債

key					lock					result
[[0, 0, 0], [1, 0, 0], [0, 1, 1]]	[[1, 1, 1], [1, 1, 0], [1, 0, 1]]	t

に答える

import copy

def expend_lock(lock, N, M, size):
    expended_lock = [[0 for i in range(size)] for _ in range(size)]
    for y in range(N):
        for x in range(N):
            expended_lock[y + M - 1][x + M - 1] = lock[y][x]

    return expended_lock

def rotate(key):
    # *key는 껍질(Unpacking)을 벗겨내는 역할을 합니다. [] () {}
    # [(0, 0, 0), (1, 0, 0), (0, 1, 1)] *key 
    # [(0, 1, 0), (0, 0, 1), (0, 0, 1)] zip(*key)
    # [(0, 1, 0), (1, 0, 0), (1, 0, 0)] reversed
    # [[0, 1, 0]] ... -> list
    # 순수 함수
    return [list(reversed(i)) for i in zip(*key)]

def is_unlock(y, x, key, lock, N, M):
    _lock = copy.deepcopy(lock)
    for _y in range(M):
        for _x in range(M):
            _lock[_y + y][_x + x] += key[_y][_x]

    for _y in range(N):
        for _x in range(N):
            if _lock[_y + M - 1][_x + M - 1] != 1:
                return False

    return True

def solution(key, lock):
    N = len(lock)
    M = len(key)
    size = (N - 1) * 2 + M
    expended_lock = expend_lock(lock, N, M, size)

    for _ in range(4):
        for y in range(size - M + 1):
            for x in range(size - M + 1):
                if is_unlock(y, x, key, expended_lock, N, M):
                    return True
        key = rotate(key)

    return False