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

행렬 테두리 회전하기

문제 설명

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부터 시작하므로 전체 배열 크기를 1 늘려서 만들었다. 그리고 숫자를 넣을 때는 (1, 1)부터 넣으면 된다.

board = [[0] * (columns + 1) for _ in range(rows + 1)]
cnt = 1
​
for i in range(1, rows + 1):
  for j in range(1, columns + 1):
    board[i][j] = cnt
    cnt += 1

회전을 시작하기전 (x1, y1)의 값은 따로 저장해둔다.

for문을 돌면서 왼쪽 세로부터 이동 시켜준다.

for row in range(x1 + 1, x2 + 1):
  min_value = min(min_value, board[row][y1])
  board[row - 1][y1] = board[row][y1]

다음으로 맨 밑줄을 이동시켜준다.

for col in range(y1 + 1, y2 + 1):
  min_value = min(min_value, board[x2][col])
  board[x2][col - 1] = board[x2][col]

다음으로 오른쪽 세로를 이동시켜준다.

for row in range(x2 - 1, x1 - 1, -1):
  min_value = min(min_value, board[row][y2])
  board[row + 1][y2] = board[row][y2]

다음으로 맨 윗줄을 이동시켜준다.

for col in range(y2 - 1, y1 - 1, -1):
  min_value = min(min_value, board[x1][col])
  board[x1][col + 1] = board[x1][col]

(x1, y1 +1) 자리에 값이 반복되므로 따로 저장해뒀던 원래 들어가야할 (x1, y1)의 값을 넣어준다.

board[x1][y1 + 1] = tmp

각 과정을 진행하면서 최솟값을 업데이트 시켜주고 회전이 한번 끝날 때마다 해당 최솟값을 answer에 넣어주면 된다.

 

전체 코드 

def solution(rows, columns, queries):
    answer = []
    board = [[0] * (columns + 1) for _ in range(rows + 1)]
    cnt = 1
    
    for i in range(1, rows + 1):
        for j in range(1, columns + 1):
            board[i][j] = cnt
            cnt += 1
    
    
    for x1, y1, x2, y2 in queries:
        tmp = board[x1][y1]
        min_value = tmp
        # 왼쪽 세로
        for row in range(x1 + 1, x2 + 1):
            min_value = min(min_value, board[row][y1])
            board[row - 1][y1] = board[row][y1]
        # 맨 밑 줄 
        for col in range(y1 + 1, y2 + 1):
            min_value = min(min_value, board[x2][col])
            board[x2][col - 1] = board[x2][col]
        # 오른쪽 세로
        for row in range(x2 - 1, x1 - 1, -1):
            min_value = min(min_value, board[row][y2])
            board[row + 1][y2] = board[row][y2]
        # 맨 윗 줄
        for col in range(y2 - 1, y1 - 1, -1):
            min_value = min(min_value, board[x1][col])
            board[x1][col + 1] = board[x1][col]
        
        board[x1][y1 + 1] = tmp
        answer.append(min_value)
    return answer