leetcode 689. 三个无重叠子数组的最大值

leetcode 689. 三个无重叠子数组的最大值

题目采用动态规划求解，设dp[i][j]为前i个数组元素组成的j个无重叠子数组最大值。那么该如何计算dp[i][j]呢？

1. 最大值不包含nums[i]，也就是dp[i-1][j]
2. 最大值包含num[i]，那么dp[i][j] = dp[i-k][j-1] + nums[i-k+1]+…+nums[i]，意思就是前i-k个元素中j-1的最大值，加上nums[i-k+1]…num[i]（这里可以采用前缀数组优化）。

此时dp[n-1][3]就是最大值，可以跟据这个回溯找到三个数组的下标。但问题是，题目要求找到字典序最小的下标。这里有两种方案，

1. 反向dp，即从数组最后往前推导。
2. 将数组反转，然后从前往后推导。

 vector<int> maxSumOfThreeSubarrays(vector<int>& nums, int k) {
        int sum[20010]={0};//前缀和
        reverse(nums.begin(),nums.end());
        for(int i=1;i<=nums.size();i++){
            sum[i]=sum[i-1]+nums[i-1];
        }
        int dp[20010][4]={0};
        for(int i=k;i<=nums.size();i++){
            for(int j=1;j<4;j++){
                dp[i][j]=max(dp[i-1][j],dp[i-k][j-1]+sum[i]-sum[i-k]);
            }
        }
        vector<int>res;
        res.assign({0,0,0});
        int i=nums.size();
        int j=3;
        int id=0;
        while(j>0){
            if(dp[i-1][j]>dp[i-k][j-1]+sum[i]-sum[i-k]){
                i--;
            }else{
                res[id++]=nums.size()-i;
                i-=k;
                j--;
            }
        }
        return res;
    }