BOJ_S1_13335

📁 [S1_13335] 트럭

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

public class Main{
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine(), " ");
        
        // 다리를 건너는 트럭의 수
        int n = Integer.parseInt(st.nextToken());
        
        // 다리의 길이
        int w = Integer.parseInt(st.nextToken());
        
        // 다리의 최대 하중
        int L = Integer.parseInt(st.nextToken());
        
        // 다리의 큐
        Queue<Integer> bridgeQueue = new LinkedList<>();
        
        // 트럭의 큐
        Queue<Integer> truckQueue = new LinkedList<>();

        st = new StringTokenizer(br.readLine()," ");

        // 트럭 넣기
        for(int i=0; i<n; i++){
            truckQueue.add(Integer.parseInt(st.nextToken()));
        }

        // 0으로 다리 초기화 시키기
        for(int i=0; i<w; i++){
            bridgeQueue.add(0);
        }

        // 총 시간
        int time = 0;

        // 다리위의 총 무게
        int bw =0;

        // 다리가 빌 때까지
        while(!bridgeQueue.isEmpty()){
            // 시간 추가
            time++;

            // 맨 마지막 것 빼기
            bw -= bridgeQueue.poll();

            // 트럭이 아직 남았다면
            if(!truckQueue.isEmpty()) {
                // 조건에 맞으면 넣기
                if (bw + truckQueue.peek() <= L) {
                    // 다리위 트럭 무게 더하기
                    bw += truckQueue.peek();
                    // 다리에 트럭 올리기
                    bridgeQueue.add(truckQueue.poll());
                } else {
                    // 한 칸 씩 밀기
                    bridgeQueue.add(0);
                }
            }
        }
        System.out.println(time);
    }
}

🤔 나의 생각

Queue의 FIFO 특성을 이용하면 접근을 쉽게 할 수 있는 것 같다.
트럭 큐와 다리 큐를 생성해 처음에 다리 큐를 0으로 채운 뒤 트럭이 들어올 때 마다 처리를 진행한다.
다리위에 총 들어올 수 있는 트럭개수는 다리 큐의 전체 크기로 처리를 해주고
전체 무게가 넘으면 안되는 것은 트럭이 남았을 때 처리를 해주며 트럭을 다리에 추가한다.