[Codetree] 컨베이어 벨트 Python

https://www.codetree.ai/missions/2/problems/conveyor-belt/description

코드트리 | 코딩테스트 준비를 위한 알고리즘 정석

 

import sys
from collections import deque

input = sys.stdin.readline

n, t = map(int, input().rstrip().split(" "))

graph = []

for i in range(2):
    temp = list(map(int, input().rstrip().split(" ")))
    graph.append(temp)


for k in range(t):
    temp = [x[-1] for x in graph]

    for i in range(2):
        for j in range(n-1, 0, -1):
            graph[i][j] = graph[i][j-1]

    graph[0][0] = temp[-1]
    graph[1][0] = temp[0]



for row in graph:
    print(*row)