[python] 프로그래머스 - 다리를 지나는 트럭

문제링크 :https://programmers.co.kr/learn/courses/30/lessons/42583

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

 

 문제설명 

 

solution 함수의 매개변수로 다리 길이 bridge_length, 다리가 견딜 수 있는 무게 weight, 트럭별 무게 truck_weights가 주어집니다. 이때 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 return 하는 solution 함수.

 

 알고리즘 

 

1. 현재 다리위에 있는 트럭의 무게와 들어가야할 트럭의 무게를  더해서 트럭이 더 올라 갈수있으면 올린다

2. 못올라가면 0 을 다리 리스트에 추가한다. 

3. while문을 트럭이 다 지나갈때 까지 돌려서 while 문이 돌아간 횟수를 return 한다.

 

 코드 

# 다리를 지나는 트럭
def solution(bridge_length, weight, truck_weights):
    answer = 0
    bridge=[]
    bridge_weight=0
    idx = 0
    bridge_length -=1
    while(1):
 
            
        if bridge_weight + truck_weights[idx] > weight:
            
            bridge.append(0)
        else:
            bridge.append(truck_weights[idx])
            bridge_weight+=truck_weights[idx]
            idx+=1
        if len(bridge) > bridge_length:
            bridge_weight-=bridge.pop(0)    
 
        answer +=1
        
        if idx==len(truck_weights):
            answer+=bridge_length
            answer+=1
            break  
    
    
    return answer