[코드트리] 컨베이어 벨트 - 파이썬

코드트리

컨베이어 벨트

https://www.codetree.ai/missions/2/problems/conveyor-belt?&utm_source=clipboard&utm_medium=text

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

 

문제

코드

• 윗 라인의 마지막, 아래 라인의 첫 번째 값을 저장하는 위치를 잘 설정해야함

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

line_top = list(map(int, input().split()))
line_bot = list(map(int, input().split()))

for _ in range(t):
    top_last = line_top[-1] 
    
    for i in range(n - 1, 0, -1):
        line_top[i] = line_top[i - 1]
    line_top[0] = line_bot[-1]  

    
    for j in range(n - 1, 0, -1):
        line_bot[j] = line_bot[j - 1]
    line_bot[0] = top_last


print(*line_top)
print(*line_bot)