[프로그래머스 코딩테스트] 다리를 지나는 트럭(Java)

  문제 설명

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

import java.util.Queue;
import java.util.LinkedList;
class Solution {
    public int solution(int bridge_length, int weight, int[] truck_weights) {
        Queue<Integer> queue = new LinkedList<>();
        int answer = 0;
        int sum = 0;
        for (int i = 0; i < truck_weights.length; i++) {
            while (true) {
                if (queue.isEmpty()) {
                    queue.add(truck_weights[i]);
                    sum += truck_weights[i];
                    answer++;
                    break;
                } else if (queue.size() == bridge_length) {
                    sum -= queue.poll();
                } else {
                    if (sum + truck_weights[i] <= weight) {
                        queue.add(truck_weights[i]);
                        sum += truck_weights[i];
                        answer++;
                        break;
                    } else {
                        queue.add(0);
                        answer++;
                    }
                }
            }
        }
        answer += bridge_length;
        return answer;
    }
}

  제네릭 타입으로 Integer를 갖는 Queue를 생성합니다. 편의를 위해 LinkedList로 구현합니다.

  변수 answer와 sum을 생성합니다.

  for문이 truck_weights의 길이만큼 순회합니다.

  whlie문은 true일 때 반복합니다.

  바깥 if문에서 queue가 비어있을 때 add 메소드를 이용하여 queue에 truck_weights의 i 번째 원소를 삽입합니다.

  sum에 truck_weights의 i 번째 원소를 더하기 할당합니다.

  반복할 때마다 answer가 증가합니다.

  break를 이용하여 if문을 종료하고 다음 반복이 진행됩니다.

  else if에서 queue의 size가 bridge_length와 같다면 queue의 맨 앞 원소에서 추출한 값을 sum에서 빼기 할당합니다.

  else의 안쪽 if문에서 sum과 truck_weigths의 i 번째 원소를 더한 값이 weight보다 작거나 같다면 다리 위에 다른 트럭이 더 올라가도 된다는 뜻입니다. 따라서 현재 다리에 있는 트럭을 가리키는 queue에 add메소드를 이용하여 truck_weights의 i 번째 원소를 추가합니다.

  다리 위에 트럭이 추가되었으므로 sum에 truck_weights의 i 번째 원소를 더하기 할당합니다.

  answer가 증가합니다.

  만약 트럭이 다리 위에 추가로 올라갈 수 없는 상황이라면 queue에 0을 삽입합니다. 이는 시간의 흐름을 상징합니다. 일정 시간이 지나 앞선 트럭이 다리를 지나가서 다른 트럭이 다리를 지나갈 수 있는 상황이 올 때까지 삽입될 것입니다.

  answer가 증가합니다.

  마지막으로 다리에 올라탄 트럭이 다리를 온전하게 지나는 시간을 계산합니다. answer에 bridge_length를 더하기 할당합니다.