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

삼각형 컨베이어 벨트 - 파이썬

문제

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

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

 

 

코드

• 삼각형의 각 변의 다른 변으로 이동하는 값을 temp 인자에 따로 저장
• temp에 저장하는 위치에 따라 결과가 바뀌니 해당 코드의 위치가 중요

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

top_left = list(map(int, input().split()))
top_right = list(map(int, input().split()))
bot = list(map(int, input().split()))

for _ in range(t):
    temp_top_left_last = top_left[-1]
    temp_top_right_last = top_right[-1]
    temp_bot_last = bot[-1]

    # top_left를 오른쪽으로 이동
    for i in range(n - 1, 0, -1):
        top_left[i] = top_left[i - 1]
    
    # top_right를 오른쪽으로 이동
    for j in range(n - 1, 0, -1):
        top_right[j] = top_right[j - 1]
    top_right[0] = temp_top_left_last

    # bot를 오른쪽으로 이동
    for k in range(n - 1, 0, -1):
        bot[k] = bot[k - 1]
    bot[0] = temp_top_right_last

    # top_left의 첫 번째 요소를 bot의 마지막 요소로 변경
    top_left[0] = temp_bot_last

print(*top_left)
print(*top_right)
print(*bot)