LeetCode——410. 分割数组的最大值(二分查找)
程序员文章站
2022-03-16 11:22:45
...
题目描述:
解题思路:
hard就是hard,看题解都看了好久,这道题是这样的,可以利用二分来做,要使最大值尽可能最小,这种套路一般可以用二分法。
我们可以思考一下,这个满足条件的数的范围,它一定是大于等于数组的最大值,小于数组值的和,那么我们就在这个范围去进行二分。
如何定条件呢,假设我们取mid,然后根据这个mid的值,从数组一开始进行分组,一直加,只要数组超过了mid我们就把之前的定为一组数组,然后从这个超过的元素在开始分,这样最后我们得到了几个数组 ,那么我们在根据这个得到的数组的个数k和题目的m去比较。如果k大于m说明数组多了,也就是我们定的这个mid他太小了,那么我们在往mid右边去二分让left=mid+1; 相反如果k小于m说明分的数组少了,我们给的这个mid’大了,那么就让right=mid,去左边寻找。这样最终就能得到答案。
这里遇到个二分的坑。如果循环条件是while(left < right)就是右值点取不到,那么在二分的时候就是 right = mid;left = mid + 1;,如果是while(left <= right)
就是right = mid-1;left = mid + 1;二分虽然简单,还是有好多坑。
二分的坑
参考代码:
public static int splitArray(int[] nums, int m) {
int left = 0, right = 0;
for (int i = 0; i < nums.length; i++) {
right += nums[i];
if (left < nums[i]) {
left = nums[i];
}
}
while (left < right) {
int mid = (right - left) / 2 + left;
if (check(nums, mid, m)) {//true的话,说明cnt<=m,分的数组少了,说明给的这个中值大了,(要加好多次),我们就让right=min
right = mid;//right取不到
} else {//false,说明cnt>m,说明给的中值这个数小了,分的数组多了。
left = mid + 1;
}
}
return left;
}
//分组判断
public static boolean check(int[] nums, int x, int m) {
int sum = 0;
int cnt = 1;
for (int i = 0; i < nums.length; i++) {
if (sum + nums[i] > x) {
cnt++;
sum = nums[i];
} else {
sum += nums[i];
}
}
return cnt <= m;
}