lintcode 43. Maximum Subarray III

给定一个整数数组和一个整数 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];
    }
}