[Programmers] level2 - 행렬 테두리 회전하기 (Python) : 2021 Dev-Matching(상반기)

▶행렬 테두리 회전하기 (Python) : 2021 Dev-Matching 상반기 (level 2)

▶문제

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 함수를 완성해주세요.

 

▶제한 사항

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

▶입출력 예제

입출력 예 #1

회전을 수행하는 과정을 그림으로 표현하면 다음과 같습니다.

입출력 예 #2

회전을 수행하는 과정을 그림으로 표현하면 다음과 같습니다.

입출력 예 #3

이 예시에서는 행렬의 테두리에 위치한 모든 칸들이 움직입니다. 따라서, 행렬의 테두리에 있는 수 중 가장 작은 숫자인 1이 바로 답이 됩니다.

 

▶풀이

행렬의 필요한 부분을 회전해서, 회전하는 숫자 중 가장 작은 녀석들의 모임을 return 해주면 된다.

일단 queries에서 할 때마다 범위를 뽑아온다.

 

다음으로 내가 한 방식은 윗줄 아랫줄 / 왼쪽 오른쪽으로 나눠서 for문을 돌렸다.

1. 윗줄은 오른쪽으로 한 칸씩만 옮겨주면 된다.

가장 왼쪽 값은 유지가 될 텐데, 어차피 왼쪽 줄을 위로 한 칸씩 올리면서 해결된다.

 

2. 아랫줄은 왼쪽으로 한 칸씩 옮겨주면 된다.

가장 오른쪽 값 역시 똑같은데, 오른쪽 줄을 아래로 한 칸씩 내리면 해결된다.

 

이렇게 오른쪽 왼쪽도 같은 방식으로 해주었다.

헷갈렸던 것은 좌표가 (0, 0)부터 시작하는 것이 아니라 (1, 1)부터 시작하는 것으로 되어있어서,

조금 헷갈리게 했던 거 같다.

그래도 큰 문제가 되지는 않아서 금방 문제를 풀었다.

def solution(rows, columns, queries):        
    answer = [[ i + j * columns + 1 for i in range(columns)] for j in range(rows)]
    result = []
                
    for r1, c1, r2, c2 in queries:
        Min = 999999999
        ans = [item[:] for item in answer]
        for i in range(c1, c2):
            ans[r1 - 1][i] = answer[r1 - 1][i - 1] #윗줄
            ans[r2 - 1][i - 1] = answer[r2 - 1][i] #아랫줄
            Min = min(Min, answer[r1 - 1][i - 1], answer[r2 - 1][i])
        for i in range(r1, r2):
            ans[i][c2 - 1] = answer[i - 1][c2 - 1] #오른쪽
            ans[i - 1][c1 - 1] = answer[i][c1 - 1] #왼쪽
            Min = min(Min, answer[i - 1][c2 - 1], answer[i][c1 - 1])
        answer = ans
        result.append(Min)
    return result

이번 2022 kakao tech internship에 있는 행렬과 연산 문제를 풀다가 이 문제를 풀어봤다.

위에 문제는 시간 초과가 자꾸 떠서, 이 문제를 해결하고 다시 풀어보기 위해서 이번 문제를 풀었다.

 

아쉽게도 이 문제는 효율성 테스트가 없어서... 카카오 문제를 해결하기엔 이번 문제가 도움이 되지 않았다.

다시 카카오 문제를 도전해봐야겠다.