행렬 테두리 회전하기

HeeSeong·2021년 6월 24일
0

프로그래머스

목록 보기
80/97
post-thumbnail

🔗 문제 링크

https://programmers.co.kr/learn/courses/30/lessons/77485


❔ 문제 설명


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입니다.

  • 모든 회전은 순서대로 이루어집니다.

  • 예를 들어, 두 번째 회전에 대한 답은 첫 번째 회전을 실행한 다음, 그 상태에서 두 번째 회전을 실행했을 때 이동한 숫자 중 최솟값을 구하면 됩니다.



💡 풀이 (언어 : Java & Python)


구현이 인덱스 때문에 상당히 복잡한 문제였다. 핵심은 시작점은 따로 보관해주고 왼쪽변부터 시작해서 꼬리부분 변 순서대로 따라가면서 돌려주는 것이다. 그러면 밀려서 사라지는 값은 이미 앞에서 옮겨져있어서 따로 값을 보관해주지 않아도 된다. 마지막에 시작점을 들어갈 위치에 넣어주고 각각의 변들의 연산마다 나오는 수들을 최소값 체킹을 하면 된다.

Java

class Solution {
    public int[] solution(int rows, int columns, int[][] queries) {
        // 정답 배열 생성
        int[] answer = new int[queries.length];
        // 초기 행렬 원소 채워서 만들기
        int[][] matrix = new int[rows][columns];
        for (int i = 0; i < rows; i++) {
            int tmp = i * columns;
            for (int j = 0; j < columns; j++)
                matrix[i][j] = tmp + j + 1;
        }
        // 행렬 테두리 회전하기
        for (int j = 0; j < queries.length; j++) {
            int[] q = queries[j];
            // 왼쪽 위 기준 숫자 보관 변수, 최소값 변수
            int keep = matrix[q[0]-1][q[1]-1], min = keep;
            // 왼쪽 변
            for (int i = q[0]; i < q[2]; i++) {
                int element = matrix[i][q[1]-1];
                matrix[i-1][q[1]-1] = element;
                min = Math.min(min, element);
            }
            // 아래쪽 변
            for (int i = q[1]; i < q[3]; i++) {
                int element = matrix[q[2]-1][i];
                matrix[q[2]-1][i-1] = element;
                min = Math.min(min, element);
            }
            // 오른쪽 변
            for (int i = q[2]-2; i >= q[0]-1; i--) {
                int element = matrix[i][q[3]-1];
                matrix[i+1][q[3]-1] = element;
                min = Math.min(min, element);
            }
            // 위쪽 변
            for (int i = q[3]-2; i >= q[1]-1; i--) {
                int element = matrix[q[0]-1][i];
                matrix[q[0]-1][i+1] = element;
                min = Math.min(min, element);
            }
            // 보관한 원소 해당 위치 삽입
            matrix[q[0]-1][q[1]] = keep;
            // 최소값 정답 배열에 넣기
            answer[j] = min;
        }
        return answer;
    }
}

Python

def solution(rows, columns, queries):
    matrix = [[0 for _ in range(columns)] for _ in range(rows)]
    answer = []
    # 행렬에 1씩 증가하는 수 채워넣음
    for i in range(rows):
        for j in range(columns):
            matrix[i][j] = i * columns + j + 1
            
    for x1, y1, x2, y2 in queries:
        # 시작점은 첫번째 for문에서 값이 사라지므로 보관
        start = matrix[x1-1][y1-1]
        # 시작점은 옮기는 수들의 min 연산에 포함되지 않으므로 포함위해 초기화
        minimum = start
        # 왼쪽 변
        for i in range(x1, x2):
            number = matrix[i][y1-1]
            matrix[i-1][y1-1] = number
            minimum = min(minimum, number)
        # 아래쪽 변
        for j in range(y1, y2):
            number = matrix[x2-1][j]
            matrix[x2-1][j-1] = number
            minimum = min(minimum, number)
        # 오른쪽 변    
        for i in range(x2, x1, -1):
            number = matrix[i-2][y2-1]
            matrix[i-1][y2-1] = number
            minimum = min(minimum, number)
        # 위쪽 변
        for j in range(y2, y1, -1):
            number = matrix[x1-1][j-2]
            matrix[x1-1][j-1] = number
            minimum = min(minimum, number)
        # 시작점은 아직 옮기지 못했으므로 여기서 반영     
        matrix[x1-1][y1] = start
        answer.append(minimum)
    
    return answer
profile
끊임없이 성장하고 싶은 개발자

0개의 댓글