[백준] 2343 - 기타 레슨 (자바 Java)

[Silver I] 기타 레슨 - 2343

문제 링크

성능 요약

메모리: 89916 KB, 시간: 732 ms

분류

이분 탐색(binary_search), 매개 변수 탐색(parametric_search)

문제 설명

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

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

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

입력

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

출력

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

---

풀이

전형적인 이분탐색, 매개변수 탐색 문제였다.

블루레이 크기를 start와 end 변수로 두고, 해당 크기의 M개의 블루레이로 모든 데이터를 담을 수 있는지를 확인하였다.

 

구조는 잘 짰는데, start(가능한 최소값)을 1로 두었더니 

반례: 5 2
1 1 1 1 100
최소 길이는 주어진 길이 중 가장 긴 값으로 해야됨

위와 같은 반례가 생겼다.

start가 1인 경우 100을 4로 담을 수 있는지를 검사하지 않아서 정답이 다르게 출력되었다.

 

start는 블루레이의 길이로 가능한 길이의 최소값으로, 존재하는 영상 중에서 가장 긴 영상의 길이로 해야 한다. 그래야 1개씩 담더라도 모든 영상을 담을 수 있기 때문이다.

 

package BOJ.Silver.s1;

import java.util.Scanner;

public class BOJ_2343_기타레슨 {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int N = sc.nextInt();
        int M = sc.nextInt();
        int[] arr = new int[N];
        int start = 1;
        for (int i = 0; i < N; i++) {
            arr[i] = sc.nextInt();
            start = Math.max(arr[i], start);
        }

        /*
        반례: 5 2
        1 1 1 1 100
       
         */
        int end = 1000000000;
        int ans = Integer.MAX_VALUE;
        while (start <= end) {
            int mid = (start + end) / 2; // 강의의 길이가 mid
            // 길이가 mid인 블루레이 M개에 강의를 모두 담을 수 있는지 확인하기
            int tmp = 1; // 필요한 블루레이의 개수
            int length = 0;
            for (int i = 0; i < N; i++) {
                if (length + arr[i] <= mid) {
                    length += arr[i];
                } else {
                    tmp++;
                    length = arr[i];
                }
            }

            if (tmp <= M) {
                // M개의 블루레이로 가능하다 -> 강의의 길이 줄여보기
                ans = mid;
                end = mid - 1;
            } else {
                start = mid + 1;
            }


        }
        System.out.println(ans);

    }
}