[Python | 파이썬] 다리를 지나는 트럭 (프로그래머스 STACK/QUEUE)

[Python | 파이썬] 다리를 지나는 트럭 (프로그래머스 STACK/QUEUE)

 

현재 다리에 올라가있는 truck을 저장하는 bridge 배열을 다리에 올라갈 수 있는 길이만큼 0으로 채워 생성한다.

매초마다 bridge의 맨위의 원소는 pop된다. 만약 현재 truck_weights의 최상단에 있는 원소의 무게와 bridge의 무게의 합이 다리가 최대로 견딜 수 있는 무게인 weight보다 작거나같다면 bridge에 이 트럭을 추가한다. 그렇지 않다면 bridge에는 0을 추가한다.

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

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

프로그래머스