📘문제 설명

트럭 여러 대가 강을 가로지르는 일차선 다리를 정해진 순으로 건너려 합니다. 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 알아내야 합니다. 다리에는 트럭이 최대 bridge_length대 올라갈 수 있으며, 다리는 weight 이하까지의 무게를 견딜 수 있습니다. 단, 다리에 완전히 오르지 않은 트럭의 무게는 무시합니다.

예를 들어, 트럭 2대가 올라갈 수 있고 무게를 10kg까지 견디는 다리가 있습니다. 무게가 [7, 4, 5, 6]kg인 트럭이 순서대로 최단 시간 안에 다리를 건너려면 다음과 같이 건너야 합니다.

따라서, 모든 트럭이 다리를 지나려면 최소 8초가 걸립니다.

solution 함수의 매개변수로 다리에 올라갈 수 있는 트럭 수 bridge_length, 다리가 견딜 수 있는 무게 weight, 트럭 별 무게 truck_weights가 주어집니다. 이때 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 return 하도록 solution 함수를 완성하세요.

💡문제 풀이

처음에 이해했던 것은 bridge_length가 다리에 올라갈 수 있는 트럭 수 인데 최대로 올라갈 수 있는 트럭 수를 의미하는 것으로 이해를 해서 bridge라는 빈 리스트를 만들고 그 안에 넣고 빼면서 계산을 하였는데 절대 정답인 8이 나올 수가 없었다.
다시 이해해보니 bridge_length가 2이면 최대 2개의 트럭이 올라갈 수 있다는 것이 아니라 2개를 유지해야 하는 의미였다.

1. bridge 리스트에 bridge_length만큼 0으로 채워서 만들고 현재 무게를 확인할 수 있는 currentWeight로 확인한다.
2. bridge에서 첫 번째 요소를 pop하고 currentWeight에서 그 값만큼 뺀다.
3. currentWeight 값이랑 truck_weight의 첫 번째 요소의 값을 더한 것이 weight보다 작거나 같으면 truck_weight에서 첫 번째 요소를 pop해서 bridge에 append한다.
4. 만약에 currentWeight 값이랑 truck_weight의 첫 번째 요소의 값을 더한 것이 weight보다 크면 bridge에 0을 append한다.
5. 3-4번은 truck_weight에 값이 있으면 실행한다.
6. 1-4번은 bridge에 값이 있는 동안 실행한다.

bridge가 2, weight이 10이고 truck_weight이 [7,4,5,6]인 테스트케이스를 살펴보면 아래와 같이 된다.

🖥️코드

from collections import deque
def solution(bridge_length, weight, truck_weights):
    time = 0
    q = deque(truck_weights)
    bridge = deque([0] * bridge_length)
    currentWeight = 0
    while bridge:
        time += 1
        currentWeight -= bridge.popleft()
        if q:
            if currentWeight + q[0] <= weight:
                currentWeight += q[0]
                bridge.append(q.popleft())
            else:
                bridge.append(0)
    return time

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