공부/코딩테스트

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

ghhong 2022. 2. 16. 16:57

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

 

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

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

programmers.co.kr

 

sum을썼었는데따로가볍게계산하게끔했음

def solution(bridge_length, weight, truck_weights):
    bridge=[]
    for i in range(bridge_length):
        bridge.append(0)
    w=0
    answer = 0
    bridge_weight=0
    while(truck_weights):
        answer+=1
        w-=bridge.pop()
        bridge= [0]+bridge
        if w+truck_weights[0] <= weight:
            bridge[0]=truck_weights.pop(0)
            w+=bridge[0]
    return answer+bridge_length