BOJ 1092 배

https://www.acmicpc.net/problem/1092

문제

지민이는 항구에서 일한다. 그리고 화물을 배에 실어야 한다. 모든 화물은 박스에 안에 넣어져 있다. 항구에는 크레인이 N대 있고, 1분에 박스를 하나씩 배에 실을 수 있다. 모든 크레인은 동시에 움직인다.

각 크레인은 무게 제한이 있다. 이 무게 제한보다 무거운 박스는 크레인으로 움직일 수 없다. 모든 박스를 배로 옮기는데 드는 시간의 최솟값을 구하는 프로그램을 작성하시오.

---

입력

첫째 줄에 N이 주어진다. N은 50보다 작거나 같은 자연수이다. 둘째 줄에는 각 크레인의 무게 제한이 주어진다. 이 값은 1,000,000보다 작거나 같다. 셋째 줄에는 박스의 수 M이 주어진다. M은 10,000보다 작거나 같은 자연수이다. 넷째 줄에는 각 박스의 무게가 주어진다. 이 값도 1,000,000보다 작거나 같은 자연수이다.

---

출력

첫째 줄에 모든 박스를 배로 옮기는데 드는 시간의 최솟값을 출력한다. 만약 모든 박스를 배로 옮길 수 없으면 -1을 출력한다.

---

예제 입력 1

3
6 8 9
5
2 5 2 4 7

예제 출력 1

2

예제 입력 2

2
19 20
7
14 12 16 19 16 1 5

예제 출력 2

4

… 이하 예제 생략

---

코드

이번 문제는 정말 딱 봐도 그리디가 생각나는 문제였다.

문제에 직접적으로 어떻게 구현하라는 내용은 없지만, 그리디 밖에 생각이 나지 않는 문제여서 구현 카테고리의 문제를 푸는 기분이 들었다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {

    static int N;
    static int[] cranes; // 크레인들의 무게 제한
    static int[] boxes; // 각 크레인이 옮겨야 할 박스 수

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        N = Integer.parseInt(br.readLine());
        cranes = new int[N];
        boxes = new int[N];
        StringTokenizer st = new StringTokenizer(br.readLine());
        for (int i = 0; i < N; i++) {
            cranes[i] = Integer.parseInt(st.nextToken());
        }
        Arrays.sort(cranes);
        int M = Integer.parseInt(br.readLine());
        st = new StringTokenizer(br.readLine());

        /**
         * 박스들의 무게를 입력받고 해당 박스의 무게를 옮길 수 있는 크레인을 찾아
         * 해당 크레인이 옮겨야 할 박스의 수에 1을 추가한다.
         */
        for (int i = 0; i < M; i++) {
            int weight = Integer.parseInt(st.nextToken());
            int j = 0;
            for (; j < N && cranes[j] < weight; j++) ;

            // 어떠한 크레인도 해당 박스를 옮길 수 없으면 -1 출력
            if(j == N) {
                System.out.println(-1);
                return;
            }

            boxes[j]++;
        }

        int answer = calculateMinTime();
        System.out.println(answer);
    }

    static int calculateMinTime() {
        int minTime = 0;

        // 남은 박스가 있을 때까지 옮기며 시간을 추가
        do {
            minTime++;
        } while (loadBoxes());

        return minTime;
    }

    static boolean loadBoxes() {
        int spareCranes = 0;
        boolean left = false;

        /**
         * 가장 큰 무게 제한을 가지고 있는 크레인부터 검사한다.
         * 검사 대상 크레인이 옮겨야 할 박스가 없다면, 여분 크레인의 수를 1 추가한다.
         * 그리고 다음 크레인이 옮겨야 할 박스를 여분 크레인의 수만큼 추가로 옮긴다.
         */
        for (int i = N - 1; i >= 0; i--) {
            spareCranes++;
            int loadingBoxes = Integer.min(boxes[i], spareCranes);
            boxes[i] -= loadingBoxes;
            spareCranes -= loadingBoxes;

            // 만약 박스를 옮기고 남은 박스가 있다면, 플래그를 true로 설정
            if(boxes[i] > 0) {
                left = true;
            }
        }

        return left;
    }
}