프로그래머스 - 다리를 지나는 트럭

COCOBALL·2023년 4월 8일
0

알고리즘

목록 보기
7/37

다리를 지나는 트럭 2️⃣

문제 설명

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

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

경과 시간다리를 지난 트럭다리를 건너는 트럭대기 트럭
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][][]

따라서, 모든 트럭이 다리를 지나려면 최소 8초가 걸립니다.

solution 함수의 매개변수로 다리에 올라갈 수 있는 트럭 수 bridge_length, 다리가 견딜 수 있는 무게 weight, 트럭 별 무게 truck_weights가 주어집니다. 이때 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 return 하도록 solution 함수를 완성하세요.

제한 조건

  • bridge_length는 1 이상 10,000 이하입니다.
  • weight는 1 이상 10,000 이하입니다.
  • truck_weights의 길이는 1 이상 10,000 이하입니다.
  • 모든 트럭의 무게는 1 이상 weight 이하입니다.

문제 풀이

💡 문제풀이 순서
  1. 한번에 다리를 건널 수 있는 트럭의 개수만큼 bridge 배열의 요소를 deque로 만들어준다.
  2. 순서를 향상시키기위해 대기중인 트럭의 배열은 deque로 만들어준다.
  3. 하나의 트럭이 지나갈 때 bridge 배열을 pop하고 bridge_weigth에 빼준고 time을 1씩 증가시킨다
  4. 만약 대기 중인 트럭이 있다면 해당 트럭과 현재 다리를 지나는 트럭의 무게가 weigth보다 작거나 같다면 해당 트럭을 작업 목록에서 빼고 bridge_weigth에 해당 트럭의 무게를 더해준다.
  5. bridge에서 건널 수 있는 트럭의 개수가 변하지 않도록 다시 배열에 요소를 추가한다.
  6. 만약 weigth보다 무겁다면 bridge에 다시 배열 요소를 추가하고 트럭이 지나갈때까지 기다린다.
from collections import deque

def solution(bridge_length,weight,truck_weights):
    truck_weights = deque(truck_weights)
    bridge = deque([0 for _ in range(bridge_length)])
    time = 0
    bridge_weight = 0

    while len(bridge) != 0:
        out = bridge.popleft()
        bridge_weight -= out
        time += 1
        if truck_weights:
            if bridge_weight + truck_weights[0] <= weight:
                left = truck_weights.popleft()
                bridge_weight += left
                bridge.append(left)
            else:
                bridge.append(0)
    return time
profile
Welcome! This is cocoball world!

0개의 댓글