PROGRAMMERS 60059: 자물쇠와 열쇠

Link : programmers.co.kr/learn/courses/30/lessons/60059

---

Python

import copy
 
 
def rotation(shape):
    '''
    :param location: (위치 좌표)
    :return: 시계방향 90도 회전
    '''
    n = len(shape)
    m = len(shape[0])
 
    result = [[0] * n for _ in range(m)]
 
    for i in range(n):
        for j in range(m):
            result[j][n - i - 1] = shape[i][j]
    return result
 
 
def check(board, K, L):
    for y in range(L):
        for x in range(L):
            if board[y + K][x + K] != 1:
                return False
    return True
 
 
def solution(key, lock):
    K, L = len(key), len(lock)
    board_map = [[0 for _ in range(K * 2 + L)] for _ in range(K * 2 + L)]
    for i in range(L):
        for j in range(L):
            board_map[i + K][j + K] = lock[i][j]
 
    for _ in range(4):
        for y in range(1, K + L):
            for x in range(1, K + L):
                t_board = copy.deepcopy(board_map)
                for j in range(K):
                    for i in range(K):
                        t_board[j + y][i + x] += key[j][i]
                if check(t_board, K, L):
                    return True
        key = rotation(key)
    return False

FeedBack

1. rotation 함수를 list(zip(*shape[::-1]))로 대체 가능함.

---

 

 

 

cjw.git@gmail.com