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

---

문제 설명

트럭 여러 대가 강을 가로지르는 일차선 다리를 정해진 순으로 건너려 합니다. 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 알아내야 합니다. 다리에는 트럭이 최대 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 이하입니다.

입출력 예

bridge_length	weight	truck_weights	return
2	10	[7,4,5,6]	8
100	100	[10]	101
100	100	[10,10,10,10,10,10,10,10,10,10]	110

---

코드 풀이

function solution(bridge_length, weight, truck_weights) {
  let answer = 0;
  const trucks = truck_weights; // 사실 필요 없긴 한데... truck_weights가 너무 길어서 대체
  const onBridge = new Array(bridge_length).fill(0);
  let bridgeWeight = 0; // 다리 위에 있는 트럭들의 총 무게
  let currentTruck;
  while (true) {
    // 매 반복은 한 번의 이동이 일어난다는 뜻이므로 answer++
    answer++;
    // 만약, 이번 이동에서 트럭이 다리를 빠져나가게 된다면? => 다리 위의 무게를 그만큼 차감
    if (onBridge[0] !== 0) bridgeWeight -= onBridge[0];
    // 맨 앞의 다리를 한 칸 제거함 (왼쪽으로 모든 트럭들을 이동시킨다는 뜻)
    onBridge.shift();
    // 그렇게 왼쪽으로 한 칸 이동했는데, 다리가 다음 트럭의 무게를 수용할 수 있다면?
    if (bridgeWeight + trucks[0] <= weight) {
      currentTruck = trucks.shift(); // 다음 올라올 트럭 = 트럭 리스트의 맨 앞
      onBridge.push(currentTruck); // 다리 끝에 트럭 하나 추가
      bridgeWeight += currentTruck; // 추가된 만큼 무게 증가
    } else {
      // 무게가 이미 초과돼서 다음 트럭이 들어올 수 없으면?
      // 앞에서 이미 한 칸 당겼기 때문에 그냥 우측 끝에는 빈 칸을 의미하는 0을 추가함
      onBridge.push(0);
    }
    // 만약 다리 위의 무게가 제로가 됐다면 ? === 더 이상 올라올 트럭이 없다는 뜻이기도 함 => 반복문 종료
    if (bridgeWeight === 0) break;
  }
  return answer;
}

참고로, break 조건을 반복문의 가장 마지막에 둔 이유는, 첫 bridgeWeight 는 0으로 설정되어 있기 때문이다. 따라서 해당 break 조건을 맨 아래에 둬야 반복문이 바로 종료되지 않고 정상적으로 실행될 수 있도록 해야한다.

---

소스 코드

function solution(bridge_length, weight, truck_weights) {
  let answer = 0;
  const trucks = truck_weights;
  const onBridge = new Array(bridge_length).fill(0);
  let bridgeWeight = 0;
  let currentTruck;
  while (true) {
    answer++;
    if (onBridge[0] !== 0) bridgeWeight -= onBridge[0];
    onBridge.shift();
    if (bridgeWeight + trucks[0] <= weight) {
      currentTruck = trucks.shift();
      onBridge.push(currentTruck);
      bridgeWeight += currentTruck;
    } else {
      onBridge.push(0);
    }
    if (bridgeWeight === 0) break;
  }
  return answer;
}

---