410. 分割数组的最大值-前缀和加动态规划算法
给定一个非负整数数组 nums 和一个整数 m ,你需要将这个数组分成 m 个非空的连续子数组。
设计一个算法使得这 m 个子数组各自和的最大值最小。
示例 1:
输入:nums = [7,2,5,10,8], m = 2
输出:18
解释:
一共有四种方法将 nums 分割为 2 个子数组。
其中最好的方式是将其分为 [7,2,5] 和 [10,8] 。
因为此时这两个子数组各自的和的最大值为18,在所有情况中最小。
示例 2:
输入:nums = [1,2,3,4,5], m = 2
输出:9
示例 3:
输入:nums = [1,4,4], m = 3
输出:4
个人觉得这一题算是动态规划里比较绕脑袋的题目,博主写的时候越也是觉得很费劲
「将数组分割为 mmm 段,求……」是动态规划题目常见的问法。
本题中,我们可以令 f[i][j]表示将数组的前 i个数分割为 j段所能得到的最大连续子数组和的最小值。在进行状态转移时,我们可以考虑第 j段的具体范围,即我们可以枚举 k,其中前 k 个数被分割为 j−1 段,而第 k+1 到第 i 个数为第 j 段。此时,这 j 段子数组中和的最大值,就等于 f[k][j−1] 与 sub(k+1,i)中的较大值,其中 sub(i,j) 表示数组 nums中下标落在区间 [i,j][i,j][i,j] 内的数的和。
解题代码如下:
int splitArray(int* nums, int numsSize, int k){
int presum[numsSize];
presum[0]=nums[0];
for(int i=1;i<numsSize;i++){
presum[i]=presum[i-1]+nums[i];
}
int dp[numsSize][k];
if(k==numsSize){
int max=0;
for(int i=1;i<numsSize;i++){
max=fmax(max,nums[i]);
}
return max;
}
for(int j=0;j<k;j++){
for(int i=0;i<numsSize;i++){
if(j==0){
dp[i][j]=presum[i];
continue;
}
if(i<j){
dp[i][j]=0;
}
else{
dp[i][j]=fmax(dp[i-1][j-1],presum[i]-presum[i-1]);
for(int kz=i-1;kz>=0;kz--){//kz下标,j位nunsaize分段,i为当前段
int max=fmax(dp[kz][j-1],presum[i]-presum[kz]);
dp[i][j]=fmin(max,dp[i][j]);
}
}
}
}
return dp[numsSize-1][k-1];
}