[프로그래머스] 코딩테스트 연습 - 행렬 테두리 회전하기

코딩테스트 연습 - 행렬 테두리 회전하기 [2021 Dev-Matching: 웹 백엔드 개발(상반기)]

풀이

해당 문제는 특별한 알고리즘을 적용시키는 것보다는 구현에 집중된 문제였던 것 같습니다.

범위만큼 회전시키고 그중에서 최솟값을 찾아 answer에 넣으면 되니까요.

 

하지만 저는 테스트 케이스 3번부터 쭉 틀려서 한참을 헤맸네요...ㅎㅎ

회전이나 최솟값을 찾는 로직을 아무리 쳐다봐도 이상할 게 없었는데 왜 계속 틀리는지 이해 안 되더라고요ㅠㅠ

이유는 생각보다 간단한 곳에 있었습니다! 처음 배열의 값을 초기화할 때에 잘못된 값으로 초기화한 것이었죠..

문제 설명에서 주어지는 입출력 예 중에서 직사각형 형태의 예제는 테두리만 돌리고 최솟값이 1이니 아무런 의심이 없이 통과하겠거니 싶었는데 그게 아니었고 직접 직사각형 형태의 테스트 케이스를 만들어보고 잘못됨을 알아냈습니다.

 

Code

def solution(rows, columns, queries):
    answer = []
    coord = [[row*columns+col+1 for col in range(columns)] for row in range(rows)]
    for q in queries:
        x1, y1, x2, y2 = q[0]-1, q[1]-1, q[2]-1, q[3]-1
        min_v = coord[x2][y2]
        rl = coord[x1][y2]
        # Top board
        coord[x1][y1+1:y2+1] = coord[x1][y1:y2]
        min_v = min(min_v, min(coord[x1][y1+1:y2+1]))
        # Left board
        for left in range(x1, x2):
            coord[left][y1] = coord[left + 1][y1]
            min_v = min(min_v, coord[left][y1])
        # Bottom board
        coord[x2][y1:y2] = coord[x2][y1+1:y2+1]
        min_v = min(min_v, min(coord[x2][y1:y2]))
        # Right board
        for right in range(x2, x1+1, -1):
            coord[right][y2] = coord[right - 1][y2]
            min_v = min(min_v, coord[right][y2])

        coord[x1+1][y2] = rl
        answer.append(min(min_v, rl))
    return answer

 

문제 설명

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]