[인프런] 6-9 뮤직비디오 (결정알고리즘)

📖  문제

---

• 첫째 줄에 자연수 N(1≤N≤1,000), M(1≤M≤N)이 주어진다.
• 다음 줄에는 조영필이 라이브에서 부른 순서대로 부른 곡의 길이가 분 단위로(자연수) 주어진다.
  • 지니 레코드에서는 불세출의 가수 조영필의 라이브 동영상을 DVD로 만들어 판매하려 한다. 순서가 바뀌는 것을 우리의 가수 조영필 씨가 매우 싫어한다. 즉, 1번 노래와 5번 노래를 같은 DVD에 녹화하기 위해서는 지니 레코드 입장에서는 이 DVD가 팔릴 것인지 확신할 수 없기 때문에 이 사업에 낭비되는 DVD를 가급적 줄이려고 한다. 그리고 M개의 DVD는 모두 같은 크기여야 제조원가가 적게 들기 때문에 꼭 같은 크기로 해야 한다.
  • 고민 끝에 지니레코드는 M개의 DVD에 모든 동영상을 녹화하기로 하였다. 이때 DVD의 크기(녹화 가능한 길이)를 최소로 하려고 한다.
  • 1번과 5번 사이의 모든 노래도 같은 DVD에 녹화해야 한다. 또한 한 노래를 쪼개서 두 개의 DVD에 녹화하면 안 된다.
  • DVD에는 총 N개의 곡이 들어가는데, DVD에 녹화할 때에는 라이브에서의 순서가 그대로 유지되어야 한다.

출력 예시

⚠️  주의사항

---

• 최소 용량 크기지만 순서를 바꿔서는 안된다.
• 이분 검색을 활용

 

✍️  이해

---

/**
 * 1. N과 M입력, N개로 이루어진 배열 arr
 * 2. 이분 검색을 위한 lt = 가장 큰 값 rt = 배열의 합
 * 3. lt <= rt 일 떄까지 반복
 *  3-1. mid 는 lt + rt 의 / 2
 *  3-2. arr를 for each 반복
 *   3-2-1. sum + i > mid 값이면 cnt++, sum은 i로 초기화 ! sum에 i값 누적
 *  3-3. cnt가 M보다 작거나 같은 경우 -> answer를 mid 값으로, 오른쪽은 더 볼 필요 없으므로 rt = mid - 1;
 *  3-4. else 왼쪽 값을 볼 필요 없고 answer 누적 필요 없음 -> lt = mid + 1;
 * 4. return answer;
 */

 

✏️  풀이

---

import java.util.Arrays;
import java.util.Scanner;

public class Main {

    public int count(int[] arr, int mid) {
        int cnt = 1;
        int sum = 0;

        for (int i : arr) {
            if(sum + i > mid){
                cnt ++;
                sum = i;
            }else{
                sum += i;
            }
        }
        return cnt;
    }

    public int solution(int N, int M, int[] arr) {
        int lt = arr[N - 1];
        int rt = Arrays.stream(arr).sum();
        int answer = 0;

        while (lt <= rt) {
            int mid = (lt + rt) / 2 ;

            if (count(arr, mid) <= M) {
                answer = mid;
                rt = mid -1;
            }else lt = mid + 1;
        }
        return answer;
    }

    public static void main(String[] args) {
        Main T = new Main();

        Scanner sc = new Scanner(System.in);

        int N = sc.nextInt();
        int M = sc.nextInt();

        int[] arr = new int[N];

        for (int i = 0; i < N; i++) {
            arr[i] = sc.nextInt();
        }

        System.out.println(T.solution(N, M, arr));
    }
}

---