인프런 커뮤니티 질문&답변

해당 케이스에서 DVD 최소 크기가 2이면, DVD의 개수는 [1, 1], [1, 1], [1] 총 3개가 되서 결과가 틀어야 하는게 아닌가 싶습니다!

제가 짠 코드로는 id 1 채점이 통과되지 않아 질문드립니다.

import java.util.Scanner;

public class Main {

    public static int solution(int n, int m, int[] arr) {
        int answer = 10_001;
        int lt = 0;
        int rt = 0;

        for (int x : arr) {
            rt += x;
        }

        while (lt <= rt) {

            int mid = (lt + rt) / 2;
            int midValue = mid + 1;
            int sum = 0;
            int cnt = 1;
            for (int i = 0; i < n; i++) {
                if (sum + arr[i] > midValue) {
                    cnt++;
                    sum = arr[i];
                } else {
                    sum += arr[i];
                }
            }

            if (cnt > m) {
                lt = mid + 1;
            } else {
                rt = mid - 1;
            }

            if (cnt == m && answer > midValue) {
                answer = midValue;
            }


        }


        return answer;
    }

    public static void main(String[] args) {
        Scanner kb = new Scanner(System.in);
        int n = kb.nextInt();
        int m = kb.nextInt();
        int[] arr = new int[n];
        for (int i = 0; i < n; i++) {
            arr[i] = kb.nextInt();
        }
        System.out.print(solution(n, m, arr));
    }

}