[프로그래머스] 다리를 지나는 트럭 (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)

어려운 문제라기 보다는.. 조건을 어떻게 걸어줄지를 더 생각해야하는 문제!

  1. 일단 첫번째 트럭을 stk에 넣고
  2. idx는 다음 트럭의 위치를 가리킨다
  3. second는 현재 초를 말한다
  4. size는 전체 트럭의 크기
  5. 따라서 첫번째 트럭을 넣었으니 idx = 1, second = 1, size = size-1



이 루트를 따르면 된다!!

  1. size 전체 트럭의 크기가 0이될 때까지 돈다

  2. trk(큐)이 안비어 있다면
    1. 안에 있는거 전부 한칸씩 움직이기
    2. 다음거 들어올 수 있나 확인 (무조건 1개만 들어옴)
      1. 확인 방법은, 지금있는 wei에 다음 [idx].wei가 가능한지?
      2. 가능하면 큐에 추가
    3. 맨 앞에있는 트럭이 나갈 수 있는지 확인 (무조건 1개만 나감)
      1. 나갈 수 있다면 큐에서 제거
      2. size도 -1
  3. trk(큐)이 비어있다면
    1. 무조건 한개 적재한다
  4. 시간 증가시킨다