프로그래머스/LV2

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

chyam_eun 2025. 2. 15. 11:06

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

 

프로그래머스

SW개발자를 위한 평가, 교육, 채용까지 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr

def solution(bridge_length, weight, truck_weights):
    time,idx = 0,0 
    passing_truck=[] # 다리를 지난 트럭
    bridge_truck=[] # 다리를 건너는 트럭
    weight_sum=0 # 다리를 지나는 트럭들의 무게
    while(idx<len(truck_weights) or bridge_truck): 
        time+=1
        for i in range(len(bridge_truck)): # 건널때마다 시간을 증가시켜줌
            bridge_truck[i][1]+=1
            
        if bridge_truck and bridge_truck[0][1]>bridge_length: # 지나는 트럭이 있을때 시간이 다리길이 초과하면
            weight_sum-=bridge_truck[0][0] # 다리를 빠져나감
            passing_truck.append(bridge_truck.pop(0))
        
        if idx<len(truck_weights): 
            if len(bridge_truck)<bridge_length and truck_weights[idx]+weight_sum<=weight: # 다리를 건너는 트럭의 수가 다리길이보다 작고, 지날 트럭의 합이 작을때
                bridge_truck.append([truck_weights[idx],1]) # 다리를 건너는 트럭에 추가
                weight_sum+=truck_weights[idx] # 무게 추가해주기
                idx+=1 # 인덱스 증가
    return time