[백준/python/20055] 컨테이너 벨트 위의 로봇

문제 링크 : 컨테이너 벨트 위의 로봇

구현 문제인데 문제이해가 조금 어려웠다. 문제 설명 그대로 구현하면 되는 문제였다.

import sys
from collections import deque

input=sys.stdin.readline
n,k=map(int, input().split())
belt=deque(list(map(int, input().split())))
robot=deque([0]*n)
cnt=0

while 1:
    belt.rotate(1)  #1. 벨트회전
    robot.rotate(1)
    robot[-1]=0
    for i in range(n-2,-1,-1): #2. 가장 먼저 올라간 로봇부터
        if robot[i] and not robot[i+1] and belt[i+1]: #로봇 다음칸이 비어있고 내구성이 1이상이면(조건)
            robot[i+1]=1
            robot[i]=0
            belt[i+1]-=1
    robot[-1]=0
    if not robot[0] and belt[0]: #3. 올라가는 위치에 로봇이 없으면 로봇을 올린다
        robot[0]=1
        belt[0]-=1
    cnt+=1
    if belt.count(0)>=k: #4. 내구성이 0인 칸이 k개 이상이면
        break
print(cnt)

1. 벨트가 각 칸 위에 있는 로봇과 함께 한 칸 회전한다.
2. 가장 먼저 벨트에 올라간 로봇부터, 벨트가 회전하는 방향으로 한 칸 이동할 수 있다면 이동한다. 만약 이동할 수 없다면 가만히 있는다.
   조건 : 로봇이 이동하기 위해서는 이동하려는 칸에 로봇이 없으며, 그 칸의 내구도가 1 이상 남아 있어야 한다.
3. 올리는 위치에 있는 칸의 내구도가 0이 아니면 올리는 위치에 로봇을 올린다.
4. 내구도가 0인 칸의 개수가 K개 이상이라면 과정을 종료한다. 그렇지 않다면 1번으로 돌아간다.

순서대로 구현!!