339. 다리를 지나는 트럭

프로그래머스

1. Python

def solution(bridge_length, weight, truck_weights):
    time = 0
    q = [0] * bridge_length
    #[0, 트럭] -> [트럭,0]-> [0,0] 
    while q:
        time += 1
        q.pop(0)
        if truck_weights:
            if sum(q) + truck_weights[0] <= weight:
                q.append(truck_weights.pop(0))
            else:
                q.append(0)
    
    return time