뮤직비디오(결정알고리즘)

DVD에는 총 N개의 곡이 들어가는데, DVD에 녹화할 때에는 라이브에서의 순서가 그대로 유지되어야 한다.

순서가 바뀌는 것을 우리의 가수 조영필씨가 매우 싫어한다. 즉, 1번 노래와 5번 노래를 같은 DVD에 녹화하기 위해서는

1번과 5번 사이의 모든 노래도 같은 DVD에 녹화해야 한다. 또한 한 노래를 쪼개서 두 개의 DVD에 녹화하면 안된다.

지니레코드 입장에서는 이 DVD가 팔릴 것인지 확신할 수 없기 때문에 이 사업에 낭비되는 DVD를 가급적 줄이려고 한다.

고민 끝에 지니레코드는 M개의 DVD에 모든 동영상을 녹화하기로 하였다. 이 때 DVD의 크기(녹화 가능한 길이)를 최소로 하려고 한다.

그리고 M개의 DVD는 모두 같은 크기여야 제조원가가 적게 들기 때문에 꼭 같은 크기로 해야 한다.

첫째 줄에 자연수 N(1≤N≤1,000), M(1≤M≤N)이 주어진다.

다음 줄에는 조영필이 라이브에서 부른 순서대로 부른 곡의 길이가 분 단위로(자연수) 주어진다.

부른 곡의 길이는 10,000분을 넘지 않는다고 가정하자.

 

이분검색을 이용해서 푸는 문제

lt는 배열에서 제일 큰 값이 가능한 dvd의 길이의 최소니 최대값을

rt는 배열의 모든 값을 더한게 나올수 있는 최대 dvd 길이를 넣는다 

 

Arrays.stream(arr).sum(); <- 1차원 배열 합 구하기

Arrays.stream(arr).max().getAsInt() <- 배열에서 최대값 구하기, optionalInt 형태로 반환되기에 getAsInt를 붙여서

그냥 int형으로 바꿔줘야한다

 

문제에선 정답을 구하기 위해서 앞에부터 차례로 더햇을때 합이 mid 값 보다 작은 모음이 몇개 나오는지 세서

모음의 갯수가 mid보다 크면 더 작은 값으로도 가능하니 rt를 mid-1로 바꾸고 아니면 lt를 mid+1로 바꾸는 식으로 답을 구했다

 

import java.util.*;

class Main {

    public int count(int[] arr, int cnt){
        int count = 1, sum = 0;
        for(int x : arr){
            if(sum + x > cnt) {
                sum = x;
                count++;
            }
            else sum += x;
        }
        return count;
    }
    public int solution(int n, int m, int[] arr){
        int lt = Arrays.stream(arr).max().getAsInt(), rt = Arrays.stream(arr).sum(), mid = 0;
        int ans = 0;
        while(lt <= rt){
            mid = (lt+rt)/2;
            if(count(arr, mid) <= m){
                ans = mid;
                rt = mid -1;
            }
            else lt = mid + 1;
        }
        return ans;
    }

    public static void main(String[] args) {
            Main t = new Main();
            Scanner in = new Scanner(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();
            System.out.println(t.solution(n, m, arr));
    }
}