[Baekjoon 2343번/JAVA] 기타 레슨

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

2343번: 기타 레슨

Question

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

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

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

 

Input

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

 

Output

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

---

💡 Solution

강의 길이의 범위가 최대 10,000분이라는 점과 강의의 순서가 바뀌면 안된다는 점에서 이분 탐색 기법을 사용하기로 결정했습니다. 10,000분 이내에서 최소가 되는 블루레이 크기를 찾아야하고, 강의의 순서가 바뀌면 안된다는 것을 다른 측면으로 봤을 때 정렬된 거와 같다고 생각하였습니다. 

 

블루레이의 크기(속해있는 강의들의 총 시간)를 인덱스로 생각하며, 이때 가능한 블루레이의 크기는 가장 긴 강의의 길이 ~ 모든 강의 길이의 총합입니다. 예제에 적용하면 블루레이 크기의 범위는 9분(가장 긴 강의) ~ 45분(모든 강의 길이의 총합)이 됩니다. 

 

이제 가능한 블루레이 크기중 최소를 출력하라고 하였으므로, 블루레이의 수가 m개가 되는 시점을 lower bound를 이용하여 구할 것입니다.

 

예를 들어, start = 9, end = 45, mid = 27인 경우는 블루레이의 수가 2개입니다. 이는 m(예제의 경우, 3)보다 작거나 같으므로 블루레이 크기를 더 작게 해주기 위해 end = mid로 설정합니다. (현재는 lower bownd를 사용하므로 end는 mid - 1이 아닌 mid로 설정합니다.)

start = 9, end = 18, mid = 13인 경우엔 블루레이의 수가 5개입니다. 이는 m보다 크므로 블루레이의 크기를 더 크게 해주기 위해 start = mid + 1로 설정합니다.

 

이러한 과정을 반복하여, 블루레이의 수가 3개인 시점의 블루레이 크기를 구해줍니다.

 

💻 Code

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

public class BJ2343 {
    
    public static int[] lectures;
    public static int n, m;

    public static int lowerBound(int start, int end, int target) {
        while(start < end) {
            int mid = (start + end) / 2;
            if(getCount(mid) > target) {
                start = mid + 1;
            } else {
                end = mid;  
            }
        }
        return start;
    }

    public static int getCount(int mid) {
        int cnt = 1;
        int remain = mid;
        for(int i = 0; i < n; i++) {
            if(lectures[i] > remain) {
                remain = mid;
                cnt++;
            }
            remain -= lectures[i];
        }
        return cnt;
    }

    public static void main(String[] args) throws IOException {
    
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());

        n = Integer.parseInt(st.nextToken());
        m = Integer.parseInt(st.nextToken());
        lectures = new int[n];

        int sum = 0;
        int maxLecture = 0;

        st = new StringTokenizer(br.readLine());
        for(int i = 0; i < n; i++) {
            lectures[i] = Integer.parseInt(st.nextToken());
            sum += lectures[i];
            if(maxLecture < lectures[i]) {
                maxLecture = lectures[i];
            }
        }

        System.out.println(lowerBound(maxLecture, sum, m));
    }
}

 

---

References

• https://12bme.tistory.com/120