문제풀이/기타

* [파이썬] [스택/큐] Pro 다리를 지나는 트럭

승무_ 2022. 2. 11. 13:09

문제

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

 

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

트럭 여러 대가 강을 가로지르는 일차선 다리를 정해진 순으로 건너려 합니다. 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 알아내야 합니다. 다리에는 트럭이 최대 bridge_length대 올라갈

programmers.co.kr

 

코드

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

 

생각 정리

예시처럼 다리 length가 2라면, [0,0] 와같은 0이 들어간 리스트를 만들고
[0, 트럭] -> [트럭,0]-> [0,0] 이런식으로 트럭이 들어오고 나가게 구현