백준 알고리즘 20055번: 컨베이어 벨트 위의 로봇 파이썬 코드 <문제풀이>

 

import sys;input=sys.stdin.readline
N,K=map(int,input().split())
A=list(map(int,input().split()))
robot=[0]*(N*2)
naegoo=0

left=0
right=N-1

second=0
while naegoo<K:

    left -= 1
    if left==-1:
        left=(2*N)-1
    right -=1
    if right==-1:
        right=(2*N)-1
    robot[right]=0
    
    for i in range(left+N-1,left-1,-1):
        j=(i+1) % (2*N)
        i=i%(2*N)

        if robot[i]==1:
            if robot[j]==1:
                continue
            if A[j]!=0:
                robot[i]=0
                robot[j]=1
                A[j]-=1
                if A[j]==0:
                    naegoo+=1

    if A[left]!=0:
        A[left]-=1
        if A[left]==0:
            naegoo+=1
        robot[left] = 1
    robot[right]=0


    second+=1

print(second)

 

python3 로 제출시 시간초과가 발생한다.

pypy로 제출하면 통과

문제를 꼼꼼히 읽고 순서대로 구현해야 하는 구현문제.