[JAVA][Level2]PROGRAMMERS 다리를 지나는 트럭

큐을 이용해서 문제를 해결하는 문제

---

다리를 지나는 트럭

---

문제 설명

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

입출력 예 설명

 

주어진 문제

class Solution {
    public int solution(int bridge_length, int weight, int[] truck_weights) {
        int answer = 0;
        return answer;
    }
}

---

나의 문제풀이

import java.util.*;

class Solution {
    public int solution(int bridge_length, int weight, int[] truck_weights) {
        int answer = 0;

        Queue<Integer> queue = new LinkedList<>();

        int count = 0; // 다리를 지난 트럭의 수
        int truckWeight = 0; // 다리를 건널 트럭의 무게
        int bridgeWeight = 0; // 현재 다리 위에 올라간 트럭의 무게합
        while (true) {
            if (count == truck_weights.length && queue.isEmpty()) break;
            if (count < truck_weights.length) truckWeight = truck_weights[count];

            answer++;
            if (answer > bridge_length) {
                bridgeWeight -= queue.poll();
            }
            if (truck_weights.length > count && bridgeWeight + truckWeight <= weight) {
                queue.offer(truckWeight);
                bridgeWeight += truckWeight;
                count++;
            } else if (truck_weights.length > count) {
                queue.offer(0);
            }

        }

        return answer;
    }
}

• 큐를 이용해서 리스트를 선언한다.
• 다리를 지난 트럭의 수를 헤아릴 count를 선언한다.
• 다리를 건널 트럭의 무게를 변수 truckWeight로 선언한다.
• 현재 다리 위에 올라간 트럭의 무게합을 변수 bridgeWeight로 선언한다.
• 다리가 견딜 수 있는 무게를 변수 truckWeight로 선언한다.
• 다리를 지난 트럭의 수가 대기 트럭의 수와 동일하고 다리를 건너는 트럭의 수가 없을 때까지 반복한다.
  • 대기트럭이 있다면 가장 앞의 트럭의 무게를 truckWeight로 입력한다.
  • 경과시간을 늘린다.
  • 대기 트럭이 남아 있고, 다리가 견딜 수 있는 무게가 남아있을 경우 큐에 트럭을 추가하고 다리 위 트럭의 무게에 더해주고 대기 트럭 순서를 넘긴다.
  • 대기 트럭이 남아 있고, 다리가 견딜 수 있는 무게가 꽉 찼을 경우 트럭의 경과 시간을 체크하기 위해 임의의 수 0을 넣어 트럭의 움직임을 체크한다.
  • 경과시간이 다리의 길이보다 커지면 트럭이 다리를 다 지났기 때문에 큐에서 트럭의 무게를 뺀 뒤 다리 위 트럭의 무게를 뺀다.

트럭을 지나는 흐름을 구현하기 위해서 FIFO 방식의 큐를 활용했다.
다른 조건들은 제한사항에 따라 조건들을 생각하면 되지만, 핵심은 대기 트럭이 남아 있는 상태에서 다리가 견딜 수 있는 무게가 꽉 찼을 경우, 빈 공간을 0으로 채워 무게에는 영향을 안 주면서 트럭이 건너가는 걸 구현하는 것 같다.
다른 여러 방식을 생각을 해봤을 때 트럭마다 이동을 구현하는데 막혔는데 결국 큐로 해서 구현에 성공했다.
자료구조를 여러 가지를 잘 알아야 그때그때 유리한 자료구조가 생각나면 간단하게 풀 수 있을듯하다.

 

 

문제 출처 - https://school.programmers.co.kr/learn/courses/30/lessons/42583?language=java