[백준] 배열 돌리기1(16926) - python

문제

크기가 N×M인 배열이 있을 때, 배열을 돌려보려고 한다. 배열은 다음과 같이 반시계 방향으로 돌려야 한다.

A[1][1] ← A[1][2] ← A[1][3] ← A[1][4] ← A[1][5]
   ↓                                       ↑
A[2][1]   A[2][2] ← A[2][3] ← A[2][4]   A[2][5]
   ↓         ↓                   ↑         ↑
A[3][1]   A[3][2] → A[3][3] → A[3][4]   A[3][5]
   ↓                                       ↑
A[4][1] → A[4][2] → A[4][3] → A[4][4] → A[4][5]

예를 들어, 아래와 같은 배열을 2번 회전시키면 다음과 같이 변하게 된다.

1 2 3 4       2 3 4 8       3 4 8 6
5 6 7 8       1 7 7 6       2 7 8 2
9 8 7 6   →   5 6 8 2   →   1 7 6 3
5 4 3 2       9 5 4 3       5 9 5 4
 <시작>         <회전1>        <회전2>

배열과 정수 R이 주어졌을 때, 배열을 R번 회전시킨 결과를 구해보자.

입력

첫째 줄에 배열의 크기 N, M과 수행해야 하는 회전의 수 R이 주어진다.

둘째 줄부터 N개의 줄에 배열 A의 원소 Aij가 주어진다.

출력

입력으로 주어진 배열을 R번 회전시킨 결과를 출력한다.

제한

• 2 ≤ N, M ≤ 300
• 1 ≤ R ≤ 1,000
• min(N, M) mod 2 = 0
• 1 ≤ Aij ≤ 108

예제 입력 1 

4 4 2
1 2 3 4
5 6 7 8
9 10 11 12
13 14 15 16

예제 출력 1 

3 4 8 12
2 11 10 16
1 7 6 15
5 9 13 14

예제 입력 2 

5 4 7
1 2 3 4
7 8 9 10
13 14 15 16
19 20 21 22
25 26 27 28

예제 출력 2 

28 27 26 25
22 9 15 19
16 8 21 13
10 14 20 7
4 3 2 1

예제 입력 3 

2 2 3
1 1
1 1

예제 출력 3 

1 1
1 1

---

import sys
input = sys.stdin.readline

def rotation(arr):
    for i in range(min(N, M)//2):
        x, y = i, i
        key = arr[x][y]   #맨 처음 좌표가 가진 값
         
        for j in range(i+1, N-i):  #왼쪽
            x = j
            tmp = arr[x][y]
            arr[x][y] = key
            key = tmp
        
        for j in range(i+1, M-i): # 아래
            y = j
            tmp = arr[x][y]
            arr[x][y] = key
            key = tmp
        
        for j in range(i+1, N-i): #오른쪽
            x = N-j-1
            tmp = arr[x][y]
            arr[x][y] = key
            key = tmp
        
        for j in range(i+1, M-i): #위
            y = M-j-1
            tmp = arr[x][y]
            arr[x][y] = key
            key = tmp
            
    return arr

N,M,R = map(int, input().split())
arr = [[0]*M for i in range(N)]

for i in range(N):
    values = list(map(int, input().split()))
    arr[i] = values

for _ in range(R):
    result = rotation(arr)

for i in range(N):
    for j in range(M):
        print(result[i][j], end =' ')
    print()

시간초과 떴다.

deque에 rotate()라는 회전 함수를 사용해야되는 것 같다. 

구현은 간단한데 생각하는게 어려워서 나중에 다시 풀어볼거다.