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

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

import sys

input = sys.stdin.readline

bridge_length, weight = map(int, input().split())
truck_weights = list(map(int, input().split()))


def solution(bridge_length, weight, truck_weights):
    bridge = [0] * bridge_length
    time = 0
    while bridge:
        time += 1
        bridge.pop(0)
        if truck_weights:
            if sum(bridge) + truck_weights[0] <= weight:
                bridge.append(truck_weights.pop(0))
            else:
                bridge.append(0)
    return time


print(solution(bridge_length, weight, truck_weights))

 

해설

다리 길이만큼 0으로 초기화 된 리스트를 만든다. 1초가 지날 때마다 리스트의 맨 앞 값을 빼준다. 이 후, 리스트와 truck_weights의 첫 번째 값의  합이 weight보다 작거나 같을 때, truck_weight의 첫 번째 값을 리스트에 넣어준다. 만약 조건에 맞지 않는다면 0을 넣어준다.