题目描述:
给定一个非负整数数组和一个整数 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,在所有情况中最小。
方法1:动态规划
主要思路:
(1)dp[ i ][ j ]表示前 i 个元素组成 j 组的情况下,可以获得各自和的最大值的最小值;
(2)则dp[ i ][ j ]的状态可以从 dp[ k ][ j-1 ]的状态下获得,既将 前 k 个元素分成 j-1 个组,后面的 i-k 个作为第 j 组,来获得可能的 dp[ i ][ j ],则此时的可能的最大和为max(dp[k][j-1],sub_sum[i]-sub_sum[k]),其中 sub_sum [ i ]表示前 i 个元素的和,然后在这些可能的选项中选出最小值即可,既dp[i][j]=min(dp[i][j],max(dp[k][j-1],sub_sum[i]-sub_sum[k]));
(3)最后,返回结果值 dp[size_nums][m];
class Solution {
public:
int splitArray(vector<int>& nums, int m) {
int size_nums=nums.size();
//动态数组
vector<vector<long long >> dp(size_nums+1,vector<long long>(m+1,LLONG_MAX));
//数据的部分和
vector<long long> sub_sum(size_nums+1,0);
for(int i=0;i<size_nums;++i){
sub_sum[i+1]=nums[i]+sub_sum[i];
}
dp[0][0]=0;//初始值
//前i个元素进行分组
for(int i=1;i<=size_nums;++i){
//最多可分的组数是 i 和 m 中的较小值
for(int j=1;j<=min(i,m);++j){
//前面可以分 j-1 组的情形
for(int k=j-1;k<i;++k){
//求最大值中的最小值
dp[i][j]=min(dp[i][j],max(dp[k][j-1],sub_sum[i]-sub_sum[k]));
}
}
}
return dp[size_nums][m];
}
};