행렬 테두리 회전하기 (lv2)

문제 설명

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입니다.
  • 모든 회전은 순서대로 이루어집니다.
  • 예를 들어, 두 번째 회전에 대한 답은 첫 번째 회전을 실행한 다음, 그 상태에서 두 번째 회전을 실행했을 때 이동한 숫자 중 최솟값을 구하면 됩니다.

---

 제일 좋은 풀이법이라고 생각되는것은, 처음 생각했던것 처럼 좌표의 숫자들을 실제로 회전하며 최소값을 계속 갱신하는 것이다. 아래링크에 해당 방법을 자세히 적어놨으며, deque모듈의 rotate 메서드를 이용한 풀이 역시 남겨놓는다.

[프로그래머스 lv2] 행렬 테두리 회전하기 (파이썬) (tistory.com)

[프로그래머스 lv2] 행렬 테두리 회전하기 (파이썬)

from collections import deque
def solution(rows, columns, queries):
    arr = [[i+columns*j for i in range(1,columns+1)] for j in range(rows)]
    answer, result = deque(), []
    for i in queries:
        a,b,c,d = i[0]-1,i[1]-1,i[2]-1,i[3]-1
        for x in range(d-b):
            answer.append(arr[a][b+x])
        for y in range(c-a):
            answer.append(arr[a+y][d])
        for z in range(d-b):
            answer.append(arr[c][d-z])
        for k in range(c-a):
            answer.append(arr[c-k][b])
        answer.rotate(1)
        result.append(min(answer))
        for x in range(d-b):
            arr[a][b+x] = answer[0]
            answer.popleft()
        for y in range(c-a):
            arr[a+y][d] = answer[0]
            answer.popleft()
        for z in range(d-b):
            arr[c][d-z] = answer[0]
            answer.popleft()
        for k in range(c-a):
            arr[c-k][b] = answer[0]
            answer.popleft()
    return result

 처음에 위와 같은 방법으로 접근했는데, 큐에 회전시킬 테두리 부분들을 넣은 후 회전시키고 나서 또 다시 그 결과를 원래 보드에 적용시키는것이 번거롭게 느껴져 다른 방법을 시도하게 되었다. 구역을 나눠 시행하는것 까지는 같으나, 한칸씩 회전시킨다는것을 큐로 접근한다는 발상이 아닌 두 좌표를 서로 스왑해나간다는 것으로 접근했다. 이렇게하면 회전과 적용이 동시에 이루어져 코드가 번거롭게 길어지지 않는다는 장점이 있다고 생각한다. 최소값을 찾는것은, 위의 경우는 어차피 테두리에 들어갈 원소들을 따로 저장해두기 때문에 최소값을 한번에 찾지만, 나는 회전 연산이 끝난 후 따로 테두리 원소들을 가져와 최소값을 찾았다.

def solution(rows, columns, queries):
    answer = []
    board = [[0]*(columns+1)] + [[0]+[columns*i + j+1 for j in range(columns)] for i in range(rows)]
    for query in queries:
        luRow,luCol,rdRow,rdCol = query
        for i in range(luRow,rdRow):
            board[i][luCol],board[i+1][luCol] = board[i+1][luCol],board[i][luCol]
        for j in range(luCol,rdCol):
            board[rdRow][j],board[rdRow][j+1] = board[rdRow][j+1],board[rdRow][j]
        for i in range(rdRow,luRow,-1):
            board[i][rdCol],board[i-1][rdCol] = board[i-1][rdCol],board[i][rdCol]
        for j in range(rdCol,luCol+1,-1):
            board[luRow][j],board[luRow][j-1] = board[luRow][j-1],board[luRow][j]
        s1 = set([board[i][j] for i in range(luRow,rdRow+1) for j in range(luCol,rdCol+1)])
        s2 = set([board[i][j] for i in range(luRow+1,rdRow) for j in range(luCol+1,rdCol)])
        answer.append(min(s1-s2))
    return answer