[백준] 16926번: 배열 돌리기1(Python)

문제

https://www.acmicpc.net/problem/16926

16926번: 배열 돌리기 1

 

문제 풀이

• 회전하는 행렬의 인덱스들을 1차원 배열로 뽑아 놓는다.(반시계방향으로)
• ex.

n, m, r = 2, 3, 2
array = [[1, 2, 3], [4, 5, 6]]

[(0, 0), (1, 0), (1, 1), (1, 2), (0, 2), (0, 1)] #위치 인덱스를 담은 배열

 

• 회전수 만큼 더해준 위치의 원소는 이동된 위치를 나타낸다.
• ex.  (0+R) => 2 #R: 2
• array[0][0] 원소는 2번째 위치인 (1,1)로 이동한다.

코드

from collections import deque
import sys
input = sys.stdin.readline

n, m, r = map(int, input().split())
array = [] #input 배열
for _ in range(n):
    array.append(list(map(int, input().split())))
    
result = [[0] * m for _ in range(n)] #결과 담는 배열

# 시작점만 주 대각선으로 잡기
row = min(n, m) // 2
start = deque([(i, i) for i in range(row)])

indices = [] #회전하는 행렬의 인덱스를 담는 배열
dx = [1, 0, -1, 0]
dy = [0, 1, 0, -1]
while start:
    temp = []
    x, y = start.popleft()

    for i in range(4):
        if i % 2 == 0:
            for j in range(n-1):
                temp.append((x, y))
                x += dx[i]
                y += dy[i]
        else:
            for j in range(m-1):
                temp.append((x, y))
                x += dx[i]
                y += dy[i]

    indices.append(temp)
    n -= 2
    m -= 2

for i in range(len(indices)):
    length = len(indices[i])
    for j in range(length):
        x, y = indices[i][j]
        nx, ny = indices[i][(j+r) % length]
        result[nx][ny] = array[x][y]

for r in result:
    print(*r)