410. 分割数组的最大值(leetcode)
题目描述
给定一个非负整数数组 nums 和一个整数 k ,你需要将这个数组分成 k 个非空的连续子数组。
设计一个算法使得这 k 个子数组各自和的最大值最小。
示例1
输入:nums = [7,2,5,10,8], k = 2
输出:18
解释:
一共有四种方法将 nums 分割为 2 个子数组。
其中最好的方式是将其分为 [7,2,5] 和 [10,8] 。
因为此时这两个子数组各自的和的最大值为18,在所有情况中最小。
示例2
输入:nums = [1,2,3,4,5], k = 2
输出:9
示例3
输入:nums = [1,4,4], k = 3
输出:4
提示信息
1 <= nums.length <= 1000
0 < = n u m s [ i ] < = 1 0 6 0 <= nums[i] <= 10^6 0<=nums[i]<=106
1 <= k <= min(50, nums.length)
看到最大值最小值,往二分答案这方面思考
题解1(C++版本)
class Solution {
public:
bool check(int x, vector<int>& nums, int k){
int sum = 0,cnt=1;
for(int i = 0; i < nums.size(); i++){
if(sum + nums[i] <= x) sum += nums[i];
else {
sum = nums[i];
cnt++;
}
}
return cnt <= k;
}
int splitArray(vector<int>& nums, int k) {
int m = 0, sum = 0;
for(int i = 0; i < nums.size(); i++) {
if(nums[i] > m) m = nums[i];
sum += nums[i];
}
int L = m - 1, R = sum + 1, mid; // 二分答案
while(L + 1 < R){
mid = (L + R)/2;
if(check(mid, nums, k)) R = mid;
else L = mid;
}
return R;
}
};