给定一个非负整数数组和一个整数 m,你需要将这个数组分成 m 个非空的连续子数组。设计一个算法使得这 m 个子数组各自和的最大值最小。
输入:
nums = [7,2,5,10,8]
m = 2
输出:
18
解释:
一共有四种方法将nums分割为2个子数组。
其中最好的方式是将其分为[7,2,5] 和 [10,8],
因为此时这两个子数组各自的和的最大值为18,在所有情况中最小。
来源:力扣(LeetCode)410
链接:https://leetcode-cn.com/problems/split-array-largest-sum
class Solution {
public int splitArray(int[] nums, int m) {
int lo = 0;
int hi = 0;
int mid = 0;
for (int i = 0;i < nums.length;i++) {
lo = nums[i] > lo ? nums[i] : lo;
hi += nums[i];
}
while (lo < hi) {
mid = lo + (hi - lo) / 2;
int pieces = splits(nums,mid);
if (pieces > m) {
lo = mid + 1;
}else {
hi = mid;
}
}
return lo;
}
private int splits(int[] arr,int num) {
int cnt = 1;
int sum = 0;
for (int i = 0;i < arr.length;i++) {
if (sum + arr[i] > num) {
cnt++;
sum = arr[i];
}else {
sum += arr[i];
}
}
return cnt;
}
}
解题思路 :二分 + 贪心, 首先要知道二分猜的是什么?猜的是子数组的最大值,这样就可以确定初始lo = max(arr[i]) hi = sum(arr[i])
范围 ,为什么呢?假如一个元素一个组,那么最大值是max(arr[i])
假如不分割,那么最大值sum(arr[i])
,对于一个非负数组来说,分的组数越多,最大值就越小,分的组数越少,最大值越大,所有起始子数组的最大值取值范围就是上面的lo和hi
二分的过程中始终使arr分割出来的子数组最大值<=mid
,分割出最小的组数pieces
,如果最小的组数pieces都比m大,说明mid选小了,将mid变大,如果最小的组数pieces<= m,mid可大可小,是不能确定mid该变大还是变小的,但是有一种贪心策略,就是让mid小一点,这样子数组的最大值也变小,这样是我们要求的答案。
这里面的贪心策略,渣渣不会证明它的正确性,就是mid变大的一部分不会得到最大值的最小值。