【二分】leetcode LCP 12. 小张刷题计划

题目

为了提高自己的代码能力，小张制定了 LeetCode 刷题计划，他选中了 LeetCode 题库中的 n 道题，编号从 0 到 n-1，并计划在 m 天内按照题目编号顺序刷完所有的题目（注意，小张不能用多天完成同一题）。

在小张刷题计划中，小张需要用 time[i] 的时间完成编号 i 的题目。此外，小张还可以使用场外求助功能，通过询问他的好朋友小杨题目的解法，可以省去该题的做题时间。为了防止“小张刷题计划”变成“小杨刷题计划”，小张每天最多使用一次求助。

我们定义 m 天中做题时间最多的一天耗时为 T（小杨完成的题目不计入做题总时间）。请你帮小张求出最小的 T是多少。

输入：time = [1,2,3,3], m = 2
输出：3
解释：第一天小张完成前三题，其中第三题找小杨帮忙；第二天完成第四题，并且找小杨帮忙。这样做题时间最多的一天花费了 3 的时间，并且这个值是最小的。

输入：time = [999,999,999], m = 4
输出：0
解释：在前三天中，小张每天求助小杨一次，这样他可以在三天内完成所有的题目并不花任何时间。

思路

1、先找出总用时的最大值和最小值
2、对最大值和最小值开始二分查找一个最小值，使数组可以划分为m组
3、在划分组时，注意可以忽略一个最大用时（最大用时就求助外援）

代码

class Solution {
public:

    bool check(vector<int>& time, int m, int k){
        int cnt = 1, sum = 0, maxx = 0;//cnt记录组数 sum记录每组的时间和 maxx记录每组的最大用时

        for(int i=0; i<time.size(); i++){
            int tmp = min(maxx, time[i]);   //用tmp排除计算最大值
            if(tmp+sum>k){
                cnt++;
                sum = 0;
                maxx = time[i];
            }else{
                sum += tmp;
                maxx = max(maxx, time[i]);
            }
        }
        return cnt <= m;
    }

    int minTime(vector<int>& time, int m) {
    	//left记录最小值 right记录最大值
        int left = 0, right = 0, n = time.size();

        if(n<=m)
            return 0;
		//最大值就是全部用时加起来
        for(int i=0; i<n; i++)
            right += time[i];

		//二分查找
        while(left<=right){
            int mid = (left + right)>>1;

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

        return left;
    }
};