[Programmers/파이썬] 프로그래머스(Lv.2) 행렬 테두리 회전하기 문제풀이

문제 설명

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

 

사고 흐름

1. row와 column 길이 제한 (100)와 query 길이 제한 (10000)을 보니 완전 탐색, 즉 진짜로 배열을 시계 방향으로 돌려야 할 것 같다....

2. 하나씩 시계 방향으로 복사하자 

3. 시계 방향으로 미는건 맞지만, 그것을 시계 방향으로 복사하면 안됨!

 

문제 핵심과 알고리즘

완전 탐색 + 좌표 이동

 

def solution(rows, columns, queries):
    answer = []
    l = [[i+j+1 for j in range(columns)] for i in range(0, rows*columns, columns)]
    
    for q in queries :
        x1, y1, x2, y2 = q[0]-1,q[1]-1,q[2]-1,q[3]-1
        last = l[x1][y1]
        m = [l[x1][y1]]
        for i in range(x1, x2) :
            l[i][y1] = l[i+1][y1]
            m.append(l[i+1][y1])
        for i in range(y1, y2) :
            l[x2][i] = l[x2][i+1]
            m.append(l[x2][i+1])
        for i in range(x2, x1, -1) :
            l[i][y2] = l[i-1][y2]
            m.append(l[i-1][y2])
        for i in range(y2, y1+1, -1) :
            l[x1][i] = l[x1][i-1]
            m.append(l[x1][i-1])
        l[x1][y1+1] = last
        answer.append(min(m))
            
    return answer

 

우선 초기의 2차원 배열을 만든다. (원래 초기 배열을 만드는 코드가 6줄 정도 됐는데, 조금 간단하게 수정했다.) 이제 query를 돌면서 배열을 하나씩 민다. 이때 시계 방향으로 한칸씩 민다고해서 복사도 시계 방향으로 복사를 하면 안된다! ㅇㅇ예를 들어 (x1, y1)를 (x1,y1+1)로 복사하고, (x1,y1+1)를 (x1,y1+2)로 복사하면, 처음에 (x1,y1+1) 값이 이미 (x1,y1) 로 바꼈기 때문에 진짜 (x1,y1+1)를 잃어버린다. 그래서 복사는 시계 반대 방향으로 진행해야 한다. (자신의 값을 유지하기 위해서) 

그리고 최소값을 찾기 위해 이동하는 모든 값을 배열 m 에 넣고 최솟값을 추출한다. 

 

문제 풀면서 헤맸던 부분

1. 최솟값 찾기 (min)

처음에는 배열 m을 사용하지 않고, m = min(m, l[x][y]) 이런 식으로 했다. 근데 왜인지 TC가 1,2번 빼고 다 틀려서,, 도저히 틀린 부분을 못찾겠어서 이 부분을 고쳤더니 다 맞았다. 왜 틀린거지?.. 추측하건데, 뭔가 값을 하나씩 이동하는거라서 한개정도 중복으로 들어가던가 안들어갔을 것 같다.. 그래도 스스로 문제점을 찾고 고쳐서 다행이다. 

 

총평 

시간이 오래 걸렸던 문제였다. 아이디어는 쉬웠지만, 구현이 좀 까다로웠다. 시간이 걸렸던 부분은 1) 진짜로 배열에서 원소를 직접 다 이동시켜야하나? 꼼수는 없을까? 2) 시계 반대 방향으로 복사하는 것 구현하기 (for문의 rangeㅠ) 3) 최솟값 찾기 에서 시간이 걸렸다. 이렇게 구현이 까다로운 문제는 차분하게 그림을 그리던가 범위(엣지)와 조건을 정리하고 풀자!!!!!!!