[プログラマー]橋を渡るトラック

質問リンク

解答方法

コード#コード#

def solution(bridge_length, weight, truck_weights):
    sec = 0
    on_bridge = []
    truck_weights = truck_weights[::-1]     # [-1] 요소를 pop 하기 위해 reverse 해서 사용

    # 대기 중이거나 다리를 건너는 트럭이 있을 경우
    while truck_weights or on_bridge:
        sec += 1
        # 다리를 건너는 트럭이 있고, 가장 첫 트럭이 다리를 다 건넜다면 pop
        if on_bridge and on_bridge[0][1] <= sec:
            on_bridge.pop(0)
        # 대기 중인 트럭이 있고, 다리 길이 및 무게 조건을 만족했을 경우 append
        if truck_weights and len(on_bridge) + 1 <= bridge_length and sum([truck[0] for truck in on_bridge]) + truck_weights[-1] <= weight:
            on_bridge.append([truck_weights.pop(), sec + bridge_length])

    return sec