[프로그래머스] 다리를 지나는 트럭

😎코딩테스트 연습>스택/큐>다리를 지나는 트럭

📘 문제풀이

def solution(bridge_length, weight, truck_weights):
    sum_weight = 0
    time = 0 #총 경과시간
    queue = [] #다리를 건너는 트럭 경과 시간
    truck = [] #다리를 건너는 트럭
    
    while truck_weights:
        time += 1
        if sum_weight + truck_weights[0] <= weight:
            sum_weight += truck_weights[0]
            q = truck_weights.pop(0)
            truck.append(q)
            queue.append(0)

        for i in range(len(queue)):
            queue[i] += 1
        if len(queue) !=0 and queue[0] == bridge_length:
            sum_weight -= truck[0] 
            queue.pop(0)
            truck.pop(0)
    
    answer = time + bridge_length #마지막 트럭은 큐 들어오자마자 끝나기 때문에 더해줌        
    return answer