LCP 12. 小张刷题计划 （ 二分扩展 ）

LeetCode：LCP 12. 小张刷题计划

二分查找， 上限 -> 一天完成，数组总和， 下限 -> 每天都交给场外，0

这题与经典二分不一样： 有一次场外求值的机会

维护一个今天所完成题目的时耗最大值，每次时间超出枚举的最小 time时，可以场外求组 -> 将所完成的题目中最费时间的交给场外。

AC Code

class Solution {
    public int minTime(int[] time, int m) {
        int len = time.length;

        // right = Arrays.stream(time).sum();  --> 15ms 5.65%
        int left = 0, right = Integer.MAX_VALUE;  // 8ms 95.56%
        while(left <= right) {
            int mid = left + (right - left) / 2;

            if(check(time, m, mid)) {
                right = mid - 1;
            } else {
                left = mid + 1;
            }
        }

        return left;
    }


    public boolean check(int[] time, int m, int target) {
        int days = 1;
        int mx = 0, total = 0, len = time.length;
        boolean yang = true;

        for(int i = 0; i < len; i++) {
            if(mx < time[i]) mx =  time[i];
            total += time[i];

            if(total > target) {
                if(yang) {
                    // 场外求助
                    yang = false;
                    total -= mx;
                } else {
                    // 第二天
                    days++;
                    // 刷新场外
                    yang = true;
                    total = 0;
                    mx = 0;
                    i--; // 刚刚这题还没做，回去再做
                }
            }
        }

        return days <= m;
    }

}