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

 

문제 출처 : programmers.co.kr/learn/courses/30/lessons/42583

코딩테스트 연습 - 다리를 지나는 트럭

 

def solution(bridge_length, weight, truck_weights):
    answer = 0 # 걸리 시간
    bridge = [0] * bridge_length

    while bridge :
        answer += 1 # 시간 1초씩 증가
        bridge.pop(0)

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

    return answer