[Python] 자물쇠와 열쇠 - 2020 카카오 공채

문제 바로가기

문제

자물쇠 영역 내에서는 열쇠의 돌기 부분과 자물쇠의 홈 부분이 정확히 일치해야 하며 열쇠의 돌기와 자물쇠의 돌기가 만나서는 안됩니다. 또한 자물쇠의 모든 홈을 채워 비어있는 곳이 없어야 자물쇠를 열 수 있습니다. 열쇠를 나타내는 2차원 배열 key와 자물쇠를 나타내는 2차원 배열 lock이 매개변수로 주어질 때, 열쇠로 자물쇠를 열수 있으면 true를, 열 수 없으면 false를 return 하도록 solution 함수를 완성해주세요.

입출력 예

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

key를 시계 방향으로 90도 회전하고, 오른쪽으로 한 칸, 아래로 한 칸 이동하면 lock의 홈 부분을 정확히 모두 채울 수 있다.

풀이

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