본문 바로가기
자바 알고리즘/백준

백준 2343 기타레슨(매개변수탐색)

by hoshi03 2024. 4. 3.

문제

강토는 자신의 기타 강의 동영상을 블루레이로 만들어 판매하려고 한다. 블루레이에는 총 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

 

• 풀이

 

다른 이분탐색 문제처럼 푸는데 몇몇 체크해야하는걸 안해서 답 통과를 못하다가 조건을 추가해서 풀었다

문제에서 강의 순서를 바꾸면 안된다고 했는데 정렬을 해버려서 정렬을 빼고

트랙 길이를 체크하는 메서드에서 트랙 자체가 m 길이보다 큰 경우를 제외하는걸 넣어주니 풀렸다..

세세한 조건같은걸 더 신경쓰면서 문제를 풀어봐야겠다

import java.io.*;
import java.util.*;


public class Main {

    static int parametric_search(int[] arr, int count){
        int l = 0;
        int n = arr.length;
        int r = 10000 * n;
        int res = -1;

        while (l <= r){
            int m = (l+r) / 2;

            if (isAble(arr, m, count)){
                res = m;
                r = m-1;
            }

            else l = m+1;
        }
        return res;
    }

    static boolean isAble(int[] arr, int maxLen, int count){
        int tmp = maxLen;
        int cnt = 1;
        for (int x : arr){
            if (maxLen < x) return false; // 이거 안해서 50%에서 걸림
            if (tmp < x){
                tmp = maxLen;
                cnt++;
            }
            tmp -= x;
        }

        return cnt <= count;
    }


    public static void main(String[] args) throws IOException {

       Scanner in = new Scanner(System.in);
       BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        int n = in.nextInt();
        int m = in.nextInt();
        int[] arr = new int[n];
        for (int i = 0; i < n; i++) arr[i] = in.nextInt();
        //Arrays.sort(arr); <- 문제 조건이 트랙 순서를 그대로 유지해야 했기에 정렬해서 순서를 바꾸면 안됬다
        System.out.print(parametric_search(arr, m));
    }
}