[ 프로그래머스 ] 2020 카카오 - 자물쇠와 열쇠

출저 : https://programmers.co.kr/learn/courses/30/lessons/60059?language=python3

# 회전 시키기 (+ 90 위치 변경)
def roll(lst):
    n = len(lst)
    change_pan = [[0]*n for _ in range(n)]
    
    for i in range(n):
        for j in range(n):
            change_pan[j][n-1-i] = lst[i][j]
    return change_pan
 
# 잠금 해제 가능 체크
def check(sX, sY, key, lock, expendSize, start ,end):
    expendlst = [[0] * expendSize for _ in range(expendSize)]
    
    # expendSize에 key 추가
    for i in range(len(key)):
        for j in range(len(key)):
            expendlst[sX+i][sY+j] += key[i][j]
            
            
    # expendList에 ock 추가 + 기존값
    for i in range(start, end):
        for j in range(start, end):
            expendlst[i][j] += lock[i - start][j - start]
            if expendlst[i][j] != 1:
                return False
    return True
    
        
        
        
def solution(key, lock):
    start = len(key) - 1  # expendList에서 lock의 시작 지점
    end = start + len(lock)  # expendList에서 lock이 끝나는 지점
    expendSize = len(lock) + start * 2  # expendList 배열의 크기
 
    # lock은 고정이고 key가 움직이는거!!!
    for a in range(0, 4):
        for i in range(end):
            for j in range(end):
                if check(i, j, key, lock, expendSize, start, end):
                    return True
        key = roll(key)
 
    return False

 

참고 : https://www.youtube.com/watch?v=RrWnBaflV2o