题目描述:
给定一个非负整数数组和一个整数 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,在所有情况中最小。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/split-array-largest-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
令f[i][j]表示将数组的前i个数分割为j段所能得到的最大连续子数组的最小值,在进行状态转移时,我们可以考虑第j段的具体范围,即我们可以枚举k,其中前k个数被分割为j-1段,而第k+1到第i个数为第j段,此时j段子数组中和的最大值,就等于f[k][j-i]与sub(k+1,i)中的较大值,其中sub(i,j)表示数组nums中下标落在区间[i,j]内数的和。
由于我们要使子数组中和的最大值最小,因此可以列出如下的状态转移方程:
对于状态f[i][j],由于我们不能分出空的子数组,因此合法的状态必须有i>=j。对于不合法的状态(i<j)的状态,由于我们的目标的和i求出最小值,因此可以将这些状态全部初始化为一个很大的数,在上述的状态转移方程中,一旦我们尝试从不合法的状态f[k][j-1]进行转移,那么max(…)将会是一个很大的数,就不会对最外层的min{…}产生任何影响。
此外,我们还需要jiangf[0][0]的值初始化为0。在上述的状态转移方程中,当j=1时,唯一的可能性就是前i个数被分成了一段。如果枚举k=0,那么就代表着这种情况; 如果k!=0,对应的状态f[k][0]是一个不合法的状态,无法进行转移。因此我们需要令f[0][0]=0。
最终的答案即为f[n][m].
class Solution {
public:
int splitArray(vector<int>& nums, int m) {
int n = nums.size();
vector<vector<long long>> f(n + 1, vector<long long>(m + 1, LLONG_MAX));
vector<long long> sub(n + 1, 0);
for (int i = 0; i < n; i++) {
sub[i + 1] = sub[i] + nums[i];
}
f[0][0] = 0;
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= min(i, m); j++) {
for (int k = 0; k < i; k++) {
f[i][j] = min(f[i][j], max(f[k][j - 1], sub[i] - sub[k]));
}
}
}
return (int)f[n][m];
}
};