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

문제 설명
트럭 여러 대가 강을 가로지르는 일차선 다리를 정해진 순으로 건너려 합니다. 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 알아내야 합니다. 다리에는 트럭이 최대 bridge_length대 올라갈 수 있으며, 다리는 weight 이하까지의 무게를 견딜 수 있습니다. 단, 다리에 완전히 오르지 않은 트럭의 무게는 무시합니다.
예를 들어, 트럭 2대가 올라갈 수 있고 무게를 10kg까지 견디는 다리가 있습니다. 무게가 [7, 4, 5, 6]kg인 트럭이 순서대로 최단 시간 안에 다리를 건너려면 다음과 같이 건너야 합니다.

제한 조건
bridge_length는 1 이상 10,000 이하입니다.
weight는 1 이상 10,000 이하입니다.
truck_weights의 길이는 1 이상 10,000 이하입니다.
모든 트럭의 무게는 1 이상 weight 이하입니다.

입출력 예
bridge_length=2
weight=10
truck_weights=[7,4,5,6]

bridge_length대만큼 다리 위에 트럭 수가 올라갈 수 있으며 이를 고려하여 weight이하까지의 무게를 견딜 수 있도록 트럭의 수를 생각해야 한다.

풀이

1. 최대 bridge_length대 만큼 올라갈 수 있으며, 다리의 길이
2. weight이하까지의 무게를 견딜 수 있음
3. 모든 트럭이 다리를 건너가는 최소 시간

from collections import deque

def solution(bridge_length, weight, truck_weights):
    answer = 0 # 시간 
    bridge = deque([0] * bridge_length) # 다리 위 상태 (다리 건너는 트럭으로 0은 빈자리)  
    truck_weights = deque(truck_weights) # 대기 트럭 큐로 변환 
    total_weight = 0 # 현재 다리 위에 있는 트럭의 총 무게 
    
    while bridge: 
        answer += 1
        
        # 다리의 맨 앞 트럭이 다리를 지나면 무게에서 제외
        total_weight -= bridge.popleft()
        
        # 대기 트럭이 있고, 다리가 견딜 수 있으면 트럭 추가
        if truck_weights:
            if total_weight + truck_weights[0] <= weight:
                truck = truck_weights.popleft() # 대기 트럭에서 하나 꺼냄
                bridge.append(truck) # 다리 위로 트럭 추가
                total_weight += truck # 현재 다리 무게 업데이트
            else: 
                # 무게 초과 시 다리 위 빈 자리 추가
                bridge.append(0)

    return answer