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

문제 링크

def solution(bridge_length, weight, truck_weights):
    answer = 0
    bridge = bridge_length * [0]

    while bridge:
        bridge.pop(0)
        answer += 1


        if truck_weights:
            head = truck_weights[0]
            if sum(bridge) + head <= weight:
                bridge.append(head)
                del truck_weights[0]
            else:
                bridge.append(0)

    return answer