알고리즘/프로그래머스

[프로그래머스] Level2, 다리를 지나는 트럭 (Python)

wookcode 2021. 6. 3. 14:25
728x90
반응형

문제

https://programmers.co.kr/learn/courses/30/lessons/42583?language=python3 

 

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

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

programmers.co.kr

 

코드

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

0    []          [0,0]      [7,4,5,6]
1    []          [0,7]      [4,5,6]
2    []          [7,0]      [4,5,6]
3    [7]        [0,4]      [5,6]
4    [7]        [4,5]      [6]
5    [7,4]      [5,0]      [6]
6    [7,4,5]    [0,6]      []
7    [7,4,5]     [6]       []
8    [7,4,5,6]    []        []

반응형