[Gold V] 16926 - 배열 돌리기1

문제

크기가 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

 

---

# 풀이 방법

처음에는 각 바이러스에서 퍼질 수 있는 영역의 수를 구한 후에 그게 큰 바이러스를 먼저 막고 이런 식으로 진행하려고 했다.

그런데 그렇게 하다보니 어떻게 기준을 세워야할지 명확하지 않았다..

여기서 생각한게 모든 수를 다 고려해야 하나..? 생각이 들었다.

 

combination을 사용하여 이를 구현해냈다.

계속 최적화하려다 보니 오히려 기본적인 코드를 놓치지 않았나

먼저 어떻게든 코드가 돌아가게 하고 그 다음 최적화를 생각하는 방향으로 진행해보자.

---

# 코드

1. Layer-by-Layer 접근으로 전환해라

• 배열은 레이어 단위( 바깥 테두리 전체)로 생각한다.
• 각 레이어는 위/오른쪽/아래/왼쪽 네 방향으로 쪼갤 수 있다.
• 방향별로 이동 규칙을 작성한다.
• 새 배열(new)을 만들어서 충돌 없이 값을 이동시킨다.
• 필요한 만큼 회전 반복한다.

# 2025-04-27 10:21-11:10
import sys
sys.stdin = open("input.txt","r")
def rotate_once(origin, N, M):
    new = [[0] * M for _ in range(N)]
    layers = min(N, M) // 2

    for layer in range(layers):
        # 왼쪽
        for i in range(layer, M - layer - 1):
            new[layer][i] = origin[layer][i + 1]
        # 위쪽
        for i in range(layer, N - layer - 1):
            new[i][M - layer - 1] = origin[i + 1][M - layer - 1]
        # 오른쪽
        for i in range(M - layer - 1, layer, -1):
            new[N - layer - 1][i] = origin[N - layer - 1][i - 1]
        # 아래쪽
        for i in range(N - layer - 1, layer, -1):
            new[i][layer] = origin[i - 1][layer]
    
    return new

def rotate(origin, N, M, R):
    for _ in range(R):
        origin = rotate_once(origin, N, M)
    return origin

def main():
    N, M, R = map(int, input().split())
    origin = [list(map(int, input().split())) for _ in range(N)]
    result = rotate(origin, N, M, R)

    for row in result:
        print(*row)

main()

---

# 참고 (너무 복잡하게 시도한 첫 풀이)

왼/아래/오/위 이렇게 따로 함수를 정의했고 돌아가는 코드도 엉망..

# 2025-04-27 10:21-11:10
import sys
sys.stdin = open("input.txt","r")

n, m, r = map(int,sys.stdin.readline().split())
graph = [list(map(int,sys.stdin.readline().split())) for _ in range(n)]
tmp = []

def go_left():
    sy = 1; ty = m-1
    for i in range(n):
        for j in range(sy+i,ty-i+1):
            graph[i][j-1] = graph[i][j]

def go_down():
    sx = n-2; tx = 0
    for j in range(m):
        for i in range(sx-j,tx+j-1,-1):
            if i == j:tmp.append((i,j,graph[i][j]))
            graph[i+1][j] = graph[i][j]

def go_right():
    sy = m-2; ty = 0
    for i in range(n-1,-1,-1):
        tmp_i = n - i
        for j in range(sy-tmp_i,ty+tmp_i-1,-1):
            graph[i][j+1] = graph[i][j]

def go_up():
    sx = 1; tx = n-1
    for j in range(m-1,-1,-1):
        tmp_j = m - j
        for i in range(sx+tmp_j,tx-tmp_j+1):
            graph[i-1][j] = graph[i][j]

def turn():
    diag = []
    for i in range(n):
        for j in range(m):
            if i == j:
                diag.append((i,j,graph[i][j]))

    go_left();go_down();go_right();go_up()

    for t in tmp:
        i,j,g = t
        graph[i+1][j] = g

def print_graph():
    for i in range(n):
        for j in range(m):
            print(graph[i][j], end=' ')
        print()

def main():
    for _ in range(r): turn()
    print_graph()

main()