프로그래머스, 다리를 지나는 트럭 (Python)

 

https://school.programmers.co.kr/learn/courses/30/lessons/42583?language=python3

프로그래머스

 

프로그래머스 알고리즘 코딩테스트 문제 다리를 지나는 트럭에 대해서 정리해보려고 한다!

 

문제 

트럭 여러 대가 강을 가로지르는 일차선 다리를 정해진 순으로 건너려 합니다. 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 알아내야 합니다. 다리에는 트럭이 최대 bridge_length대 올라갈 수 있으며, 다리는 weight 이하까지의 무게를 견딜 수 있습니다. 단, 다리에 완전히 오르지 않은 트럭의 무게는 무시합니다.

예를 들어, 트럭 2대가 올라갈 수 있고 무게를 10kg까지 견디는 다리가 있습니다. 무게가 [7, 4, 5, 6]kg인 트럭이 순서대로 최단 시간 안에 다리를 건너려면 다음과 같이 건너야 합니다.

문제 이해

트럭이 bridge_length 를 순차적으로 지날 때, 최소지간이 얼마나 걸리는지 체크하는 문제이다.

다리에는 length 와 weight 가 있어 트럭이 지날 수 있는 최대 무게가 존재하여 해당 무게가 넘으면 뒤에있는 트럭이 지나가지 못한다.

 

문제 예시를 보면 다음과 같다

0	[]	[]	[7,4,5,6]
1~2	[]	[7]	[4,5,6]
3	[7]	[4]	[5,6]
4	[7]	[4,5]	[6]
5	[7,4]	[5]	[6]
6~7	[7,4,5]	[6]	[]
8	[7,4,5,6]	[]	[]

 

어떻게 접근할 것인지

표를 보면 알 수 있듯 선입선출로 보인다. FIFO 방식이어서 queue 를 사용해야한다고 생각했고, 시작점과 끝점이 다르기때문에 dequeue 를 이용하여 삽입 O(1) 삭제 O(1) 로 해결하려고 했다.

 

결과 코드

from collections import deque

def solution(bridge_length, weight, truck_weights):
    bridge = deque([0] * bridge_length) # 다리 설정
    total_weight = 0 # 다리에 있는 트럭의 총 무게
    answer = 0
    
    while truck_weights:
        total_weight -= bridge.pop() # 1초마다 다리에 차가 지나간다. 
        if total_weight + truck_weights[0] <= weight: # 현재 순서인 트럭이 지나갈 수 있을 경우
            truck = truck_weights.pop(0) # 트럭을 빼서 O(1)
            bridge.appendleft(truck) # 다리에 올린다 O(1)
            total_weight += truck # 무게 증가 O(1)
        else: # 무게 초과인 경우
            bridge.appendleft(0) # 0 삽입. O(1)
        answer += 1
    
    answer += bridge_length
    return answer

 

처음에는 남은 트럭을 기준으로 생각해서 작업을 했는데, 그게 아니라 bridge_length 를 기준으로 작업하니 훨씬 수월했다

실제 문제 표도 기준은 bridge_length 를 선입선출로 보여주고있었는데 놓쳐서 시간이 오래걸렸다.