[ 2023-05-26 🌴 TIL ]

LV. 2 다리를 지나는 트럭

문제 링크

def solution(bridge_length, weight, truck_weights):
    time = 0
    bridge = [0 for _ in range(bridge_length)]  
    # bridge에 제한 수 만큼 선언
    bridge_w = 0 # 현재 다리 위의 트럭 무게 측정 할 때 사용
    while bridge:
        time += 1
        bridge_w -= bridge.pop(0) # 0을 넣어주기 때문에 - 가능
        if truck_weights:
          # 트럭이 다 지나가면 남은 bridge를 제거만 해준다.
            if bridge_w + truck_weights[0] <= weight:
                # 다음 트럭의 무게와 현재 다리위 트럭들의 무게 합이 제한을 
                # 넘는 지 확인
                t = truck_weights.pop(0)
                bridge.append(t)  # 다리에 트럭을 추가
                bridge_w += t
            else:
                bridge.append(0)  
                # 대기 중인 트럭이 있으면 bridge크기를 제한 길이만큼 유지

    return time

검색을 통하여 문제를 해결하였는데 무게를 확인하는 조건에서
SUM을 쓰면 테스트5번에서 시간초과가 걸리는 것을 확인하였습니다.

그래서 따로 무게를 측정하는 변수를 두어서 조건에 적용하였습니다.