给定一个非负整数数组和一个整数 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,在所有情况中最小。
链接:https://leetcode-cn.com/problems/split-array-largest-sum
思路分析:
二分法:
这道题比较难想到用二分,因为潜在的有序关系不直观。
二分枚举的题,我们首先需要思考3个问题:
- 二分枚举的相关判定变量是否具有单调性?
- l, r, mid 代表着什么?
- 用于缩小区间的判断条件是什么?
对于本题来说,我们需要求 在m次划分下 最小的 子数组各自和最大值。
令 x 为划分数,y为子数组各自和最大值。可以看出,x范围在[1,nums.size],并且x与y单调递减。这里的 x 和 y 就是相关判定变量。
当 x = 1 时,y就是整个nums数组的和,也就是 r 所代表的值。
当 x = nums.size 时,y就是nums中最大元素的值,也就是 l 所代表的值。
通过 (l+r)/2 所得到的 mid 就是枚举值,充当一个假设的已知条件。
现在,我们的已知条件是 m 和 mid. m相当于题目给出的 x = m, mid相当于我们在值域中枚举的一个数 y’. 当 x = m,y可能存在许多值,即 x=m 对应一个 y 的集合。我们的任务是求 y 集合中的最小值。
首先,我们得知道我们的 mid 与 y 的关系,然后才能根据这个关系二分缩小区间,直到 mid = y.
如何知道 mid 与 y的关系呢?前面说到 mid 相当于 y’. 我们通过mid可以求出 x’,然后让 x和x’比较。因为x与y单调递减,所以如果x’ > x 则表示 mid < y,区间需要向右缩小。这样就可以达到我们的目的了。
二分枚举的原理说明白了,现在简要说一下步骤。
- 找到二分枚举的相关判定变量 x 和 y. y是题目要求的值,此题代表子数组各自和最大值。x是可以影响到 y 的值,此题代表划分数。
- 判断 x 和 y 之间的关系。可以发现,x 和 y 单调递减。
- 通过 x 的范围算出 y 的范围,并取边界值 l 和 r 作为枚举范围。
- 算出 mid,即 y’. 再通过 y’ 算出 x’,然后通过比较 x 和 x’ 的关系得到 y 和 y’ 的关系,进而缩小枚举区间。即通过 mid 算出 cnt,然后令 cnt 和 m 比较。
- 根据题意,cnt > m,区间缩小到右半部分;cnt < m,区间缩小到左半部分;cnt = m,区间缩小到左半部分(求y集合的最小值)。
- 最后,l (或 r) 的值就是所求值。
class Solution {
public:
int splitArray(vector<int>& nums, int m) {
int n = nums.size();
long long l = nums[0], r = 0;
for(int i = 0;i < n;i++)
{
r += nums[i];
l = nums[i]>l?nums[i]:l;
}
while(l < r)
{
long long mid = (l+r)/2; //y'
int cnt = 1; //x'
long long tmp = 0;
for(int i = 0;i < n;i++)
{
tmp += nums[i];
if(tmp > mid)
{
tmp = nums[i];
cnt++; //通过y'求出x'
}
}
if(cnt > m) l = mid+1; //通过比较x与x'来缩小区间
else r = mid;
}
return l;
}
};