给定一个非负整数数组 nums
和一个整数 m
,你需要将这个数组分成 m
个非空的连续子数组。
设计一个算法使得这 m
个子数组各自和的最大值最小。
示例 1:
输入:nums = [7,2,5,10,8], m = 2 输出:18 解释: 一共有四种方法将 nums 分割为 2 个子数组。 其中最好的方式是将其分为 [7,2,5] 和 [10,8] 。 因为此时这两个子数组各自的和的最大值为18,在所有情况中最小。
示例 2:
输入:nums = [1,2,3,4,5], m = 2 输出:9
示例 3:
输入:nums = [1,4,4], m = 3 输出:4
class Solution {
public:
bool check(vector<int>& nums, int& k, int target){
int count = 1, sum = 0;
for (int num : nums){
sum += num;
if (sum > target){
count++;
sum = num;
}
if (count > k)
return false;
}
return true;
}
int splitArray(vector<int>& nums, int k) {
long long left = -1, right = 0;
for(int num : nums){
left = max(left, (long long)num);
right += num;
}
while(left < right){//左闭右开
long long mid = left + (right - left) / 2;
if (check(nums, k, mid))
right = mid;
else
left = mid + 1;//目标值还得再大点(因为已经超过题目要求的k个了)
}
return left;
}
};