BOJ 16926 - 배열 돌리기 1

문제

백준 온라인 저지 - 16926번

풀이 과정

배열을 회전하는 로직을 구현하는 문제입니다.
python 으로 제출하니 시간초과가 발생하여 pypy3 로 제출했습니다.

코드

import sys


def rotate():
    x, y = 0, 0
    n, m = N, M
    time = min(N, M) // 2

    while time:
        cache = board[x][y]

        # 윗쪽
        for i in range(m - 1):
            board[x][y + i] = board[x][y + i + 1]

        # 오른쪽
        for i in range(n - 1):
            board[x + i][y + m - 1] = board[x + i + 1][y + m - 1]

        # 아래쪽
        for i in range(m - 1):
            board[x + n - 1][y + m - 1 - i] = board[x + n - 1][y + m - 2 - i]

        # 왼쪽
        for i in range(n - 1):
            board[x + n - 1 - i][y] = board[x + n - 2 - i][y]
        board[x + 1][y] = cache

        n -= 2
        m -= 2
        x += 1
        y += 1
        time -= 1


def solution():
    for _ in range(R):
        rotate()

    for row in board:
        print(*row)


if __name__ == '__main__':
    N, M, R = list(map(int, sys.stdin.readline().split()))
    board = [list(map(int, sys.stdin.readline().split())) for _ in range(N)]

    solution()