Programmers / 2단계 / 행렬 테두리 회전하기 / python

 

 

출처: https://school.programmers.co.kr/learn/courses/30/lessons/77485

 

코드

def solution(rows, columns, queries):
    matrix = []
    tmp = []
    
    for i in range(1, (rows * columns + 1)):
        tmp.append(i)
        if (len(tmp) == columns):
            matrix.append(tmp)
            tmp = []
    rotation(matrix, queries) 
    return answer
    
answer = []
def rotation(matrix, queries):
    global answer
    for query in queries:
        l = []
        y1, x1, y2, x2 = query[0]-1, query[1]-1, query[2]-1, query[3]-1
        
        start = matrix[y1][x1]
        prev = matrix[y1][x1]
        tmp = -1
        cur_x = x1; cur_y = y1
        move = [1,0]
        
        while (tmp != start):
            cur_x += move[0]
            cur_y += move[1]
            
            tmp = matrix[cur_y][cur_x]
            matrix[cur_y][cur_x] = prev
            prev = tmp
            
            if (cur_x == x2):
                move = [0,1]
            if (cur_y == y2):
                move = [-1,0]
            if (cur_x == x1):
                move = [0, -1]
            
            l.append(tmp)
        answer.append(min(l)