[백준 문제풀이] 얼렁뚱땅 20055 컨베이어 벨트 위의 로봇 풀이

https://www.acmicpc.net/problem/20055

20055번: 컨베이어 벨트 위의 로봇

 

n, k = map(int,input().split())
arr = list(map(int,input().split()))
visited = [0] * len(arr)


circle = 0
while True:

    #한 바퀴 돈다
    circle += 1

    #로봇과 함께 한 칸 회전한다.
    arr = [arr[-1]] + arr[:-1]
    visited = [visited[-1]] + visited[:-1]

    if visited[n-1] == 1 :
        visited[n-1] = 0

    #가장 먼저 벨트에 올라간 로봇부터, 한칸 이동 할 수 있으면 이동한다.
    # index 0~n-1까지에만 로봇이 존재할 것임 여기만 보면 됨
    count = 0
    for i in range(n-1,-1,-1) :
        if visited[i] == 1 :
            if arr[i+1] > 0  and visited[i+1] == 0:
                visited[i+1] = 1
                visited[i] = 0
                arr[i+1] -= 1
    
    if visited[n-1] == 1 :
        visited[n-1] = 0
    
    #1번 자리에 로봇 올려
    if arr[0] > 0 :
        arr[0]-= 1
        visited[0] = 1

    count = 0
    for i in range(len(arr)) :
        if arr[i] == 0 :
            count += 1

    if count >= k :
        break

print(circle)