给定一个整数数组和一个整数 k,找出 k 个不重叠子数组使得它们的和最大。每个子数组的数字在数组中的位置应该是连续的。
返回最大的和。
Example
样例1
输入:
List = [1,2,3]
k = 1
输出: 6
说明: 1 + 2 + 3 = 6
样例2
输入:
List = [-1,4,-2,3,-2,3]
k = 2
输出: 8
说明: 4 + (3 + -2 + 3) = 8
Notice
子数组最少包含一个数
此题类似https://leetcode.com/problems/best-time-to-buy-and-sell-stock-iv/description/,O(N^3)容易想到,可以把时间复杂度优化到O(N^2),根据递推公式,遍历时维护一个t=dp[i-1][m]+nums[m+1]+....nums[j]的最大值(子问题+后面连续子串的和)的变量就行。
public class Solution {
/**
* @param nums: A list of integers
* @param k: An integer denote to find k non-overlapping subarrays
* @return: An integer denote the sum of max k non-overlapping subarrays
*/
public int maxSubArray(int[] nums, int k) {
int len=nums.length;
int[][] dp=new int[k][len];
for(int i=0;i<k;i++){
//t=dp[l]+sum
int t=i==0?0:dp[i-1][i-1];
for(int j=i;j<len;j++){
if(i==0){
dp[i][j]=Math.max(j==0?nums[0]:dp[i][j-1],t+nums[j]);
t=Math.max(0,t+nums[j]);
}else{
//i!=0,是否包含最后一位
dp[i][j]=Math.max(j==i?Integer.MIN_VALUE:dp[i][j-1],t+nums[j]);
t=Math.max(t+nums[j],dp[i-1][j]);
}
}
}
return dp[k-1][len-1];
}
}