[Programmers] 42583. 다리를 지나는 트럭

/**
 * bridge_length : 다리에 올라갈 수 있는 트럭 수
 * weight : 다리가 견딜 수 있는 무게
 * truck_weights : 트럭 별 무게
 */

import java.util.*;

class Solution {
    public int solution(int bridge_length, int weight, int[] truck_weights) {
        Queue<Integer> queue = new LinkedList<>();
		int sum = 0; // 현재 다리를 건너는 트럭 무게의 합
		int time = 0; // 모든 트럭이 다리를 건너는 최소의 시간

		for(int i = 0; i < truck_weights.length; i++) { // 향상된 for문을 쓰는게 좋을 것 
			int truck = truck_weights[i];

			while(true) {
				// [1] 큐가 비어있는 경우 = 어떠한 트럭도 다리위에 없는 경우
				if(queue.isEmpty()) { 
					queue.add(truck);
					sum += truck;
					time++; // 다리에 오를 때만 시간 추가 
					break;
				} else if(queue.size() == bridge_length) { // [2] 큐가 가득찬 경우
					sum -= queue.poll();
				} else  { // [3] 큐가 다리 길이만큼 가득 차지 않는 경우
					// weight 값을 넘지 않는 선에서 새로운 트럭을 다리에 올려줌 
					if(sum + truck <= weight) {
						queue.add(truck);
						sum += truck;
						time++;
						break;
					} else { 
						// weight 값을 넘는다면, 0을 넣어 이미 큐에 있는 트럭이 다리를 건너게 만듬 
						queue.add(0);
						time++;
					}
				}
			}
		}

        // 마지막 트럭에서 반복문이 끝나는데, 마지막 역시 다리 길이만큼 지나가야 하기 때문에, 다리 길이만큼 더해준다.
		return time + bridge_length; 
    }
}

// 다리를 지나는 트럭(23.12.18) - 1차 복습
import java.util.*;

class Solution {
    public int solution(int bridge_length, int weight, int[] truck_weights) {
        int sum = 0; // 현재 다리를 지나는 트럭 무게의 합
        int time = 0; // 모든 트럭이 다리를 건너는 최소 시간의 합(걸린 시간)

        Queue<Integer> bridgeQueue = new LinkedList<>(); // 다리를 지나기 전 트럭
        for(int truck : truck_weights) {
        
            while(true) {
                // 1. 큐가 비어있는 경우 트럭 추가
                if(bridgeQueue.isEmpty()) {
                    bridgeQueue.offer(truck);
                    sum += truck;
                    time++; // 다리에 오를 때만 시간 추가 
                    break;
                }
                // 2. 다리에 건너는 트럭 무게의 합 == 견딜 수 있는 최대무게(weight)
                else if(bridgeQueue.size() == bridge_length) {
                    sum -= bridgeQueue.poll();
                }

                // 3. 큐가 비어있지 않은 경우
                else {
                    // 2-2. (다음 트럭까지 포함하여) 다리에 건너는 트럭 무게의 합 > 견딜 수 있는 최대 무게
                    if(sum + truck > weight) {
                        bridgeQueue.offer(0);
                        time++;
                    }
                    // 2-3. (다음 트럭까지 포함하여) 다리에 건너는 트럭 무게의 합 < 견딜 수 있는 최대 무게
                    else {
                        bridgeQueue.offer(truck);
                        sum += truck;
                        time++;
                        break;
                    }

                }
            }
        }
        // 걸린 시간 + 마지막 트럭의 통과 시간
        return time + bridge_length;
    }
}