큐 - 다리를 지나는 트럭 / 파이썬

from collections import deque

def solution(bridge_length, weight, truck_weights):
    answer = 0
    q = [0] * bridge_length
    q = deque(q)
    truck_weights = deque(truck_weights)
    while q:
        answer += 1
        q.popleft()
        if truck_weights:
            if sum(q) + truck_weights[0] <= weight:
                front = truck_weights.popleft()
                q.append(front)
            else:
                q.append(0)
    return answer

 

다른 풀이

def solution(bridge_length, weight, truck_weights):
    time = 0
    q = [0] * bridge_length
    
    while q:
        time += 1
        q.pop(0)
        if truck_weights:
            if sum(q) + truck_weights[0] <= weight:
                q.append(truck_weights.pop(0))
            else:
                q.append(0)
    
    return time

 

programmers.co.kr/learn/courses/30/lessons/42583

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