문제 설명
트럭 여러 대가 강을 가로지르는 일차선 다리를 정해진 순으로 건너려 합니다. 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 알아내야 합니다. 다리에는 트럭이 최대 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,4,5] | [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 |
💡 내가 생각한 아이디어
문제를 통해 이해해보기
OS 강의를 듣고 있는 나.. 이 문제를 보고 대기 큐(Ready Queue)가 생각나서 Queue로 문제를 풀어야겠다는 생각은 했다.
아마 '다리를 지나는 트럭' 부분의 자료구조를 큐로 만들어 주어야겠다.
근데 대기 벡터(truck_weights)에서 뽑아갈 때가 문제였다.
'다리를 지나는 트럭' 큐는 사이즈가 최대 bridge_length였고, 대기 벡터의 front값을 뽑아갈 때, 큐에 push 해줄 수 있는 조건은 다음과 같다:
대기 벡터의 front값+현재 큐에 있는 트럭 무게의 총합이 weight를 넘지 않는다.
즉,
i) if 해당 조건 만족
✔️ 대기 벡터의 front값을 뽑아서 큐에 push해준다.
(push해주는 동시에 '다리를 지나는 트럭' 큐의 총합에 해당 front값을 더해주어야 한다는 점에 유의),
ii) else
✔️ '다리를 지나는 트럭' 큐에 0을 push해준다.
🖇️ '다리를 지나는 트럭' 큐는 매 clk마다 front에 있는 값을 뽑아낸다. ➡️ '다리를 지나는 트럭' 큐가 빌 때까지 코드 반복
| 경과 시간 | 다리를 지난 트럭 | 다리를 지나는 트럭 | 대기 트럭 |
|---|---|---|---|
| 0 | [] | [0,0] | [7,4,5,6] |
| 1 | [] | [0,7] | [4,5,6] |
| 2 | [] | [7,0] | [4,5,6] |
| 3 | [7] | [0,4] | [5,6] |
| 4 | [7] | [4,5] | [6] |
| 5 | [7,4] | [5,0] | [6] |
| 6 | [7,4,5] | [0,6] | [] |
| 7 | [7,4,5] | [6] | [] |
| 8 | [7,4,5,6] | [] | [] |
요 정도까지는 혼자 생각해냈는데,
문제는 '다리를 지나는 트럭' 큐의 사이즈가 최대 bridge_length여야 한다는 조건이었다.
구글링을 통해 결국 힌트를 얻었다.
🌟 큐의 최대 사이즈가 항상 bridge_length이려면,
맨 초기(clk=0)일 때 bridge_length 사이즈만큼 큐에 0을 밀어넣고, 각 clk 마다 front값을 뽑아가는 컨셉을 잡으면 큐가 넘칠 일이 없다.
🖥️ 정답 코드
#include <string>
#include <vector>
#include <queue>
using namespace std;
queue<int> q; // 다리를 지나는 트럭
int solution(int bridge_length, int weight, vector<int> truck_weights) {
int clk = 0;
for(int i = 0; i < bridge_length; i++){
q.push(0); // bridge_length 사이즈만큼의 큐 생성
}
int sum = 0; // 지나는 큐의 트럭 무게 총합
int idx = 0;
while(!q.empty()){ // 지나는 큐가 빌 때까지 반복
int t = q.front();
sum -= t;
q.pop(); // 현재 지나고 있는 큐의 front를 뽑아줌
if(idx < truck_weights.size()) {
if(truck_weights[idx] + sum <= weight) {
// 현재 대기 트럭의 front를 넣을 수 있는지
q.push(truck_weights[idx]);
sum += truck_weights[idx];
idx++;
}
else
q.push(0);
}
clk++;
}
return clk;
}