[Codetree] 삼각형 컨베이어 벨트 Python

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

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

 

import sys
from collections import deque

input = sys.stdin.readline

n, time = map(int, input().split(" "))

graph = []

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



for t in range(time):

    temp = deque([x[-1] for x in graph])
    temp.appendleft(temp.pop())

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

    for i in range(3):
        graph[i][0] = temp[i]
    
    
for row in graph:
    print(*row)