[Programmers][2020 KAKAO BLIND RECRUITMENT] 자물쇠와 열쇠

혼자서 열심히 풀려고 했지만 1시간 안에 푸는 게 불가능해 답을 보면서 공부하기로 했다. ㅠ

풀이 힌트
1. key 회전
2. 자물쇠 확장

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):
    # 순수 함수
    # 순수 함수가 아니면 Side Effect가 있을 수 있는 함수라고 한다.
    _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 # 최소 1개가 겹쳐야되니 N-1
    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