[python][프로그래머스] 행렬테두리 회전하기

🚀  들어가며...

• 2021 Dev-Matching: 웹 백엔드 개발자(상반기)
• '레벨 2'의 문제입니다.

 

🔗 문제  

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

코딩테스트 연습 - 행렬 테두리 회전하기

 

📑 내용

[ 문제 설명 ]

rows x columns 크기인 행렬이 있습니다. 행렬에는 1부터 rows x columns까지의 숫자가 한 줄씩 순서대로 적혀있습니다. 이 행렬에서 직사각형 모양의 범위를 여러 번 선택해, 테두리 부분에 있는 숫자들을 시계방향으로 회전시키려 합니다. 각 회전은 (x1, y1, x2, y2)인 정수 4개로 표현하며, 그 의미는 다음과 같습니다.

x1 행 y1 열부터 x2 행 y2 열까지의 영역에 해당하는 직사각형에서 테두리에 있는 숫자들을 한 칸씩 시계방향으로 회전합니다.
다음은 6 x 6 크기 행렬의 예시입니다.

이 행렬에 (2, 2, 5, 4) 회전을 적용하면, 아래 그림과 같이 2행 2열부터 5행 4열까지 영역의 테두리가 시계방향으로 회전합니다. 이때, 중앙의 15와 21이 있는 영역은 회전하지 않는 것을 주의하세요.

행렬의 세로 길이(행 개수) rows, 가로 길이(열 개수) columns, 그리고 회전들의 목록 queries가 주어질 때, 각 회전들을 배열에 적용한 뒤, 그 회전에 의해 위치가 바뀐 숫자들 중 가장 작은 숫자들을 순서대로 배열에 담아 return 하도록 solution 함수를 완성해주세요.

[ 제한 조건 ]

1. rows는 2 이상 100 이하인 자연수입니다.
2. columns는 2 이상 100 이하인 자연수입니다.
   처음에 행렬에는 가로 방향으로 숫자가 1부터 하나씩 증가하면서 적혀있습니다.
   즉, 아무 회전도 하지 않았을 때, i 행 j 열에 있는 숫자는 ((i-1) x columns + j)입니다.
3. queries의 행의 개수(회전의 개수)는 1 이상 10,000 이하입니다.
4. queries의 각 행은 4개의 정수 [x1, y1, x2, y2]입니다.
   (1) x1 행 y1 열부터 x2 행 y2 열까지 영역의 테두리를 시계방향으로 회전한다는 뜻입니다.
   (2) 1 ≤ x1 < x2 ≤ rows, 1 ≤ y1 < y2 ≤ columns입니다.
   (3) 모든 회전은 순서대로 이루어집니다.
   (4) 예를 들어, 두 번째 회전에 대한 답은 첫 번째 회전을 실행한 다음, 그 상태에서 두 번째 회전을 실행했을 때 이동한 숫자 중 최솟값을 구하면 됩니다.

[ 입출력 예시 ]

 

🔗 풀이

 

우선 rows별 columns값을 가지고 있는 matrix를 만들어 줍니다.

후에 좌측 제일 위 벨류값은 회전하면 사라지기 때문에 temp 변수에 저장하게 됩니다.

그 후 좌측, 하단, 우측, 상단 순으로 회전하고, 각 회전 후 최소값을 저장합니다. 단 우측과 상단 회전 시 벨류 값들이 더 큰 위치에서 작은 위치로 이동하기 때문에 역순으로 순회합니다.

이 후 temp에 저장했던 값(matrix상에서 사라졌던 값)을 기존 temp 위치에서 우측으로 +1 한 곳에 다시 넣어줍니다.

소스코드 및 주석을 보신다면 이해가 쉬울 것입니다.

 

💌 소스코드

def solution(rows, columns, queries):
    answer = []
    matrix = []
    for r in range(rows):       # rows별 columns값을 가지고있는 matrix를 만들어 줍니다.
        matrix.append([a for a in range(r * columns + 1, (r + 1) * columns + 1)])

    for query in queries:
        query = [x - 1 for x in query]   # 인덱스는 0부터 시작하므로 -1
        temp = matrix[query[0]][query[1]]  # 좌측 제일 위 벨류 저장 (※중요: 제일 먼저 이동하는 벨류값은 matrix에서 사라지기 때문에 임시로 넣어두고 기존자리 +1 한 곳에 다시 넣어준다.)
        min_num = temp

        # 좌측 값 회전
        for i in range(query[0] + 1, query[2] + 1):
            matrix[i - 1][query[1]] = matrix[i][query[1]]
            min_num = min(min_num, matrix[i][query[1]])     # 최소값 저장

        # 하단 값 회전
        for i in range(query[1] + 1, query[3] + 1):
            matrix[query[2]][i - 1] = matrix[query[2]][i]
            min_num = min(min_num, matrix[query[2]][i])     # 최소값 저장

        # 우측 값 회전 (※중요: 우측과 상단 시계방향으로 벨류값 이동 시 더 큰 위치에서 작은 위치로 이동하기 때문에 for문에 역순으로 순회)
        for i in range(query[2] - 1, query[0] - 1, -1):
            matrix[i + 1][query[3]] = matrix[i][query[3]]
            min_num = min(min_num, matrix[i][query[3]])     # 최소값 저장

        # 상단 값 회전 (※중요: 우측과 상단 시계방향으로 벨류값 이동 시 더 큰 위치에서 작은 위치로 이동하기 때문에 for문에 역순으로 순회)
        for i in range(query[3] - 1, query[1] - 1, -1):
            matrix[query[0]][i + 1] = matrix[query[0]][i]
            min_num = min(min_num, matrix[query[0]][i])     # 최소값 저장

        matrix[query[0]][query[1] + 1] = temp

        answer.append(min_num)

    return answer

 

🙋🏻‍♂️ 후기

레벨2 이고, 생각하기 꽤 까다로운 문제였던것 같습니다. 시간이 굉장히 오래 소요되었고, 생각의 발상이 필요했던 문제였습니다.

핵심은 좌측 제일 위의 값을 temp로 저장하여 마지막에 원래위치에 우측으로 +1 한 값에 다시 넣어주는 부분, 우측과 상단 회전 시 역순으로 순회하는 부분이었던 것 같습니다.

아직 많이 부족하다는 것을 다시 한번 느끼게 되었고, 비슷한 문제유형을 좀 더 풀어봐야겠습니다.