[프로그래머스] 다리를 지나는 트럭 (java) Queue
2020-04-14
다리를 지나는 트럭 (프로그래머스> Stack_Queue)
문제설명
트럭 여러 대가 강을 가로지르는 일 차선 다리를 정해진 순으로 건너려 합니다. 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 알아내야 합니다. 트럭은 1초에 1만큼 움직이며, 다리 길이는 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 |
풀이
package Stack_Queue;
import java.util.*;
public class prog_2_re_다리를지나는트럭 {
static class Pos{
int size;
int on_bri;
Pos(int size, int on_bri){
this.size = size;
this.on_bri = on_bri;
}
}
static public int solution(int bridge_length, int weight, int[] truck_weights) {
LinkedList<Pos> trk = new LinkedList<>();
int size = truck_weights.length;
//일단 넣고 시작
trk.add(new Pos(truck_weights[0],1)); //0들어가고 1번째르 건너고있다
int idx = 1; //다음 트럭의 위치
int second = 1; // 하나 쌓았으니까 1초 지나감
size = size-1;
while(size!=-1) {
//trk가 안비어있으면
if(!trk.isEmpty()) {
//한칸씩 다 움직여주기
for(int i=0; i<trk.size(); i++)
trk.set(i, new Pos(trk.get(i).size,trk.get(i).on_bri+1));
//그리고 다음거 들어올 수 있나 보기
if(idx<truck_weights.length) {
int wei = 0;
for(int i=0; i<trk.size(); i++)
wei += trk.get(i).size;
//지금 있는거 wei에 다음 weight 추가한게 적재 가능하면
if(wei+truck_weights[idx] <= weight) {
trk.add(new Pos(truck_weights[idx],1));
idx++;
}
}
//맨 처음게 나갈 시간이면 빼주기
if(trk.get(0).on_bri == bridge_length) {
trk.remove();
size = size-1;
}
}
//트럭이 비었으면 무조건 한개 적재!
else {
trk.add(new Pos(truck_weights[idx],1));
idx++;
}
//시간 증가시키기
second++;
}//while
return second+1;
}
public static void main(String[] args) {
int s[] = {10};
System.out.println(solution(100,100,s));
}
}
후기 (40min)
어려운 문제라기 보다는.. 조건을 어떻게 걸어줄지를 더 생각해야하는 문제!
- 일단 첫번째 트럭을 stk에 넣고
- idx는 다음 트럭의 위치를 가리킨다
- second는 현재 초를 말한다
- size는 전체 트럭의 크기
- 따라서 첫번째 트럭을 넣었으니 idx = 1, second = 1, size = size-1
이 루트를 따르면 된다!!
-
size 전체 트럭의 크기가 0이될 때까지 돈다
- trk(큐)이 안비어 있다면
- 안에 있는거 전부 한칸씩 움직이기
- 다음거 들어올 수 있나 확인 (무조건 1개만 들어옴)
- 확인 방법은, 지금있는 wei에 다음 [idx].wei가 가능한지?
- 가능하면 큐에 추가
- 맨 앞에있는 트럭이 나갈 수 있는지 확인 (무조건 1개만 나감)
- 나갈 수 있다면 큐에서 제거
- size도 -1
- trk(큐)이 비어있다면
- 무조건 한개 적재한다
- 시간 증가시킨다