[ Programmers / CodingTest / Python ] 행렬 테두리 회전하기

황승환·2022년 3월 28일
0

Python

목록 보기
256/498

문제 설명

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

입출력 예시

rows	columns	queries										result
6		6		[[2,2,5,4],[3,3,6,6],[5,1,6,3]]				[8, 10, 25]
3		3		[[1,1,2,2],[1,2,2,3],[2,1,3,2],[2,2,3,3]]	[1, 1, 5, 3]
100		97		[[1,1,100,97]]								[1]

입출력 예 설명

입출력 예 #1

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

입출력 예 #2

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

입출력 예 #3

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

접근 방법

이 문제는 4개의 라인을 따로 따로 회전시키는 반복문을 작성하여 해결하였다. 위쪽 가로를 1번, 왼쪽 세로를 2번, 아래쪽 가로를 3번, 오른쪽 세로를 4번으로 가정했을 때, 1번과 4번의 경우에는 인덱스가 큰쪽에서 작은쪽으로 나아가며 다음값과 현재값을 바꿨고, 2번과 3번의 경우에는 인덱스가 작은쪽에서 큰쪽으로 나아가며 다음값과 현재값을 바꿨다. 이 과정에서 가장 작은 값을 갱신해 나아가도록 하고, 결과적인 가장 작은 값을 반환하도록 한다.

  • 결과값을 저장할 리스트 answers를 선언한다.

  • 격자를 나타낼 리스트 grid를 선언한다.

  • rows만큼 반복하는 i에 대한 for문을 돌린다.
    -> grid에 1+i*columns부터 columns+i*columns+1까지로 이뤄진 리스트를 넣는다.

  • cycling함수를 y1, x1, y2, x2를 인자로 갖도록 선언한다.
    -> 결과값 result를 maxsize로 선언한다.
    -> x2-1부터 x1까지 반복하는 감소하는 i에 대한 for문을 돌린다.
    --> grid[y1][i]grid[y1][i+1]grid[y1][i+1]grid[y1][i]로 갱신한다.
    --> result를 result, grid[y1][i], grid[y1][i+1] 중 가장 작은 값으로 갱신한다.
    -> y1부터 y2까지 반복하는 i에 대한 for문을 돌린다.
    --> grid[i][x1]grid[i+1][x1]grid[i+1][x1]grid[i][x1]로 갱신한다.
    --> result를 result, grid[i][x1], grid[i+1][x1] 중 가장 작은 값으로 갱신한다.
    -> x1부터 x2까지 반복하는 i에 대한 for문을 돌린다.
    --> grid[y2][i]grid[y2][i+1]grid[y2][i+1]grid[y2][i]로 갱신한다.
    --> result를 result, grid[y2][i], grid[y2][i+1] 중 가장 작은 값으로 갱신한다.
    -> y2-1부터 y1까지 반복하는 감소하는 i에 대한 for문을 돌린다.
    --> grid[i][x2]grid[i+1][x2]grid[i+1][x2]grid[i][x2]로 갱신한다.
    --> result를 result, grid[i][x2], grid[i+1][x2] 중 가장 작은 값으로 갱신한다.
    -> result를 반환한다.

  • queries를 순회하는 y1, x1, y2, x2에 대한 for문을 돌린다.
    -> answers에 cycling(y1-1, x1-1, y2-1, x2-1)을 넣는다.

  • answers를 반환한다.

    solution.py

    import sys
    def solution(rows, columns, queries):
       answers=[]
       grid=[]
       for i in range(rows):
           grid.append(list(range(1+i*columns, columns+i*columns+1)))
       def cycling(y1, x1, y2, x2):
           result=sys.maxsize
           for i in range(x2-1, x1-1, -1):
               grid[y1][i], grid[y1][i+1]=grid[y1][i+1], grid[y1][i]
               result=min(result, grid[y1][i], grid[y1][i+1])
           for i in range(y1, y2):
               grid[i][x1], grid[i+1][x1]=grid[i+1][x1], grid[i][x1]
               result=min(result, grid[i][x1], grid[i+1][x1])
           for i in range(x1, x2):
               grid[y2][i], grid[y2][i+1]=grid[y2][i+1], grid[y2][i]
               result=min(result, grid[y2][i], grid[y2][i+1])
           for i in range(y2-1, y1, -1):
               grid[i][x2], grid[i+1][x2]=grid[i+1][x2], grid[i][x2]
               result=min(result, grid[i][x2], grid[i+1][x2])
           return result
       for y1, x1, y2, x2 in queries:
           answers.append(cycling(y1-1, x1-1, y2-1, x2-1))
       return answers
![](https://velog.velcdn.com/images%2Fxx0hn%2Fpost%2F7fbcee56-0fb3-4980-96d0-c9620d0f4f17%2F%E1%84%89%E1%85%B3%E1%84%8F%E1%85%B3%E1%84%85%E1%85%B5%E1%86%AB%E1%84%89%E1%85%A3%E1%86%BA%202022-03-28%20%E1%84%8B%E1%85%A9%E1%84%8C%E1%85%A5%E1%86%AB%2011.35.19.png)
profile
꾸준함을 꿈꾸는 SW 전공 학부생의 개발 일기

0개의 댓글