본문 바로가기

학습 노트/알고리즘 (Python)

99클럽 - 행렬 테두리 회전하기

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

아이디어

풀이

def solution(rows, columns, queries):
    template = [[y * columns + x + 1 for x in range(columns)] for y in range(rows)]
    answer = []

    for query in queries:
        y1, x1, y2, x2 = [i - 1 for i in query]
        
        min_value = float('inf')
        top_left = template[y1][x1]

        # 좌
        for y in range(y1, y2):
            template[y][x1] = template[y + 1][x1]
            min_value = min(min_value, template[y][x1])

        # 하
        for x in range(x1, x2):
            template[y2][x] = template[y2][x + 1]
            min_value = min(min_value, template[y2][x])

        # 우
        for y in range(y2, y1, -1):
            template[y][x2] = template[y - 1][x2]
            min_value = min(min_value, template[y][x2])

        # 상
        for x in range(x2, x1 + 1, -1):
            template[y1][x] = template[y1][x - 1]
            min_value = min(min_value, template[y1][x])

        template[y1][x1 + 1] = top_left
        min_value = min(min_value, top_left)

        answer.append(min_value)

    return answer