[Java] 백준 2343. 기타 레슨

📌 문제 링크 - https://www.acmicpc.net/problem/2343

 

 

---

 

 

 

문제

강토는 자신의 기타 강의 동영상을 블루레이로 만들어 판매하려고 한다. 블루레이에는 총 N개의 강의가 들어가는데, 블루레이를 녹화할 때, 강의의 순서가 바뀌면 안 된다. 순서가 뒤바뀌는 경우에는 강의의 흐름이 끊겨, 학생들이 대혼란에 빠질 수 있기 때문이다. 즉, i번 강의와 j번 강의를 같은 블루레이에 녹화하려면 i와 j 사이의 모든 강의도 같은 블루레이에 녹화해야 한다.

강토는 이 블루레이가 얼마나 팔릴지 아직 알 수 없기 때문에, 블루레이의 개수를 가급적 줄이려고 한다. 오랜 고민 끝에 강토는 M개의 블루레이에 모든 기타 강의 동영상을 녹화하기로 했다. 이때, 블루레이의 크기(녹화 가능한 길이)를 최소로 하려고 한다. 단, M개의 블루레이는 모두 같은 크기이어야 한다.

강토의 각 강의의 길이가 분 단위(자연수)로 주어진다. 이때, 가능한 블루레이의 크기 중 최소를 구하는 프로그램을 작성하시오.

 

입력

첫째 줄에 강의의 수 N (1 ≤ N ≤ 100,000)과 M (1 ≤ M ≤ N)이 주어진다. 다음 줄에는 강토의 기타 강의의 길이가 강의 순서대로 분 단위로(자연수)로 주어진다. 각 강의의 길이는 10,000분을 넘지 않는다.

 

출력

첫째 줄에 가능한 블루레이 크기중 최소를 출력한다.

 

 

 

 

 

 

---

 

 

풀이

이분 탐색으로 풀 수 있는 문제다.

처음에는 min 값을 1로 설정하고 풀었는데 50%에서 틀렸다. 

min 값이 작아도 이분 탐색을 하다 보면 결국 답이 나올 텐데 왜 틀릴까??

 

이 문제는 각 블루레이에 강의를 나눠 담을 때 블루레이의 크기 중 최소를 구하는 문제이다.

즉, 하나의 블루레이는 최소한 가장 긴 강의 하나는 담을 수 있어야 한다. 

 

아래의 예제를 보자.

5 4
1 5 9 3 10

강의의 개수가 5, 블루레이의 개수가 4일 때 각 강의의 길이가

1, 5, 9, 3, 10 이라면 

 

블루레이 1 : 1, 5

블루레이 2 : 9

블루레이 3 : 3

블루레이 4 : 10

 

1, 5, 9, 3, 10

이렇게 넣으면 각 블루레이의 크기는 6, 9, 3, 10이 되고 최종 블루레이의 크기는 10이다. 

 

이때 min 값을 1로 설정한다면 블루레이의 크기(mid)가 가장 긴 길이인 10보다 작게 설정되는 순간이 존재한다.

그러면 10은 블루레이의 크기보다 크므로 블루레이에 절대 담을 수 없게 된다.

하지만 코드에서는 새로운 블루레이를 만들고 진행하기 때문에 실제로는 불가능한 상황을 가능하다고 여기고 넘어간다.

 

이 때문에 불필요하게 블루레이 개수가 증가하고 정확한 계산이 이루어지지 않아 잘못된 답이 나오게 된다.

따라서 정확한 계산을 위해서는 반드시 가능한 범위에서 이분 탐색을 진행해야 하고, min 값은 강의 중 가장 긴 강의의 길이로 설정해 주거나 강의의 길이가 블루레이의 크기(mid)보다 크면 불가능하게 처리해 줘야 한다.

 

 

---

 

 

코드

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

// https://www.acmicpc.net/problem/2343
public class Main_2343 {
    private static final int MAX = 1_000_000_000;
    private static int N, M; // 강의의 수 N, 블루레이 개수 M
    private static int maxLen; // 가장 긴 강의의 길이
    private static int[] length; // 기타 강의의 길이


    public static void main(String[] args) throws IOException {
        sol();
    }//main


    private static void sol() throws IOException {
        init();

        int result = 0;
        int min = maxLen;
        int max = MAX;
        int mid;

        while(min <= max) {
            mid = (min + max) / 2;

            if(isAvailable(mid)) {
                max = mid - 1;
                result = mid;
            }
            else {
                min = mid + 1;
            }
        }

        System.out.print(result);
    }//sol


    private static boolean isAvailable(int mid) {
        int sum = 0; // 블루레이의 크기
        int cnt = 1; // 블루레이 개수

        for(int len : length) {
            //if(len > mid) return false; // min을 1로 설정하면 따로 처리해야 됨.

            if(sum + len > mid) {
                cnt++;
                sum = len;
            }
            else {
                sum += len;
            }
        }

        return M >= cnt;
    }//isAvailable


    private static void init() throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        // 강의의 수 N (1 ≤ N ≤ 100,000)과 M (1 ≤ M ≤ N)이 주어진다.
        StringTokenizer st = new StringTokenizer(br.readLine());
        N = Integer.parseInt(st.nextToken()); // 강의의 수
        M = Integer.parseInt(st.nextToken()); // 블루레이 개수

        length = new int[N]; // 기타 강의의 길이

        // 기타 강의의 길이가 강의 순서대로 분 단위로(자연수)로 주어진다.
        st = new StringTokenizer(br.readLine());
        for(int i=0; i<N; i++) {
            length[i] = Integer.parseInt(st.nextToken());
            maxLen = Math.max(maxLen, length[i]);
        }

        br.close();
    }//init


}//class