[백준 2343] 기타 레슨 (Java) - 이분탐색

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

기타 레슨

문제

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

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

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

입력

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

출력

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

예제 입력 1 복사

9 3
1 2 3 4 5 6 7 8 9

예제 출력 1 복사

17

 

---

풀이

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

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 M = Integer.parseInt(st.nextToken()); // 블루레이 개수

        int[] A = new int[N];

        int sum = 0; // 강의시간 총합
        int max = 0; // 강의 최대길이 시간
        int minSize = 0; // 블루레이 최소시간

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

        // 강의시간 입력받기, 강의길이 총합, 최대길이시간 구하기
        for(int i=0; i<N; i++){
            A[i] = Integer.parseInt(st.nextToken()); // 강의시간 입력받기
            sum += A[i];
            max = Math.max(max, A[i]);
        }

        // 이분탐색 index
        int start = max;
        int end = sum;
        int result = 0;

        // 강의 최대길이 ~ 총합 사이에서 이분탐색
        while (start <= end){
            int mid = (start + end) / 2; // 현재 블루레이 크기 후보
            int count = 1; // 필요한 블루레이 개수 (최소 1개 필요)
            int lectureSum = 0; // 현재 블루레이에 저장된 강의 길이 합

            for(int lecture : A){
                // 하나의 블루레이에 들어갈 강의시간 합계가 블루레이 하나의 크기보다 커지면
                if(lectureSum + lecture > mid){
                    count++; // 블루레이 갯수 + 1
                    lectureSum = 0; // 강의시간 합계 초기화
                }
                // 그 외에는 강의시간 총합 + 현재강의시간
                lectureSum += lecture;
            }

            // for문 다 돌고나왔을 때 필요한 블루레이 개수 비교
            if(count <= M){
                // 블루레이 개수보다 count가 작으면 블루레이 크기 줄여도 됨
                result = mid;
                end = mid - 1;
            }else{
                // count > M 이면 블루레이 크기를 늘려야 함
                start = mid + 1;
            }
        }

        System.out.println(result);
    }
}

 

이분탐색 탐색 범위

• 최소값 : 가장 긴 강의의 길이 (어떤 블루레이든 최소한 가장 긴 강의 하나는 들어가야 함)
• 최대값 : 모든 강의를 하나의 블루레이에 담았을 때의 총합

이분탐색

• mid값을 블루레이의 크기로 가정, 현재 mid 크기로 M개의 블루레이에 나눌 수 있는지 확인
• 가능하면 더 작은 크기로 가능할지 확인 (end = mid - 1)
• 불가능하면 블루레이 크기를 늘려야 함 (start = mid + 1)