Leetcode 410. 分割数组的最大值【二分法】

问题描述

给定一个非负整数数组和一个整数 m，你需要将这个数组分成 m 个非空的连续子数组。设计一个算法使得这 m 个子数组各自和的最大值最小。

注意:
数组长度 n 满足以下条件:

1 ≤ n ≤ 1000
1 ≤ m ≤ min(50, n)
示例:

输入:
nums = [7,2,5,10,8]
m = 2

输出:
18

解释:
一共有四种方法将nums分割为2个子数组。
其中最好的方式是将其分为[7,2,5] 和 [10,8]，
因为此时这两个子数组各自的和的最大值为18，在所有情况中最小。

解题报告

解释见代码。

实现代码

class Solution {
public:
    bool match(vector<int>nums, int mid, int m){
        // 注意 这里的cnt初始化为1，因为最后一个数组在这里无法计数的
        int cnt=1;
        long long sum=0;
        for(int i=0;i<nums.size();i++){
            if(sum+nums[i]>mid){
                sum=nums[i];
                cnt++;
                continue;
            }
            sum+=nums[i];
        }
        return cnt>m;
    }
    int splitArray(vector<int>& nums, int m) {
        long long l=0, r=0, mid;
        for(int i=0;i<nums.size();i++){
            if(l<nums[i]) l=nums[i];
            r+=nums[i];
        }
        // 这里为什么不能取等号呢？当l=r时，循环内match返回false的情况就不用说了，重点说明一下循环内match返回true，此时l=mid+1 r=mid，但是实际上可选的答案在【l,r】之间，此时返回的答案会超出已选的答案范围。
        while(l<r){
            // 假设每个子数组之和小于mid，得到的最少的数组个数
            mid=l+(r-l)/2;
            if(match(nums, mid, m)){
                l=mid+1;
            }
            else
                r=mid;
        }
        return l;
    }
};

参考资料

[1] Leetcode 410. 分割数组的最大值