[프로그래머스] LV.2 다리를 지나는 트럭

[프로그래머스] LV.2 다리를 지나는 트럭

 

1. 문제 : Link

건널때마다 1만큼 지나갈때마다 1초가 걸리고

다리가 지탱할 수 있는 무게만큼의 차만 지나다닐 수 있다고 했을때

최단시간으로 건널 때의 시간

 

2. 풀이

queue를 이용해서  다리위에있는 모든 차의 무게와 새로 들어오는 무게를 합하여 감당가능한 무게와 비교했다.

 

3. 코드

def solution(bridge_length, weight, truck_weights):
    bridge = [truck_weights.pop(0)]
    bridge_count = [1]
    time = 1
    while 1:
        time += 1
        if not truck_weights:
            time += bridge_length-1
            break
        else:
            if bridge_count[0] == bridge_length:
                bridge_count.pop(0)
                bridge.pop(0)
            if sum(bridge) + truck_weights[0] <= weight:
                bridge.append(truck_weights.pop(0))
                bridge_count.append(0)
        for i in range(len(bridge_count)):
            bridge_count[i] += 1
    return time

좀 더 나은 코드

def solution(bridge_length, weight, truck_weights):
    
    answer = 0
    bridge = [0 for _ in range(bridge_length)]
    
    while bridge:
        
        answer += 1
        bridge.pop(0)
        
        if truck_weights:
            if sum(bridge) + truck_weights[0] <= weight:            
                t = truck_weights.pop(0)
                bridge.append(t)
            else:
                bridge.append(0)
                 
         
    return answer