No.263 - LeetCode[689] Maximum Sum of 3 Non-Overlapping Subarrays

思路：
前缀和预处理+dp求最大和+回溯路径

1、前缀和预处理后，退化成一道，求间距k的3个数最大和（经典dp）。
2、经典3维dp求最大和。
3、记录每次变更时的当前下标，以便回溯路径。

难点：
有很多边界处理

/*
 * @lc app=leetcode id=689 lang=cpp
 *
 * [689] Maximum Sum of 3 Non-Overlapping Subarrays
 */

// @lc code=start
class Solution {
public:
    vector<int> maxSumOfThreeSubarrays(vector<int>& nums, int k) {
        int N = nums.size();
        // 前缀和
        vector<int> sum(N-k+1, 0);
        int base = 0;
        for(int i=0;i<k;i++) base += nums[i];
        for(int i=0;i<N-k;i++){
            sum[i] = base;
            base -= nums[i];
            base += nums[i+k];
        }
        sum[N-k] = base;
        // dp
        vector<vector<int> > dp(3, vector<int>(N-k+1, 0));
        vector<vector<int> > loc(3, vector<int>(N-k+1, 0));
        dp[0][0] = sum[0];
        for(int i=1;i<=N-k;i++){
            // one
            if(sum[i] > dp[0][i-1]){
                dp[0][i] = sum[i];
                loc[0][i] = i;
            }else{
                dp[0][i] = dp[0][i-1];
                loc[0][i] = loc[0][i-1];
            }
            // two
            if(i >= k){
                if(sum[i] + dp[0][i-k] > dp[1][i-1]){
                    dp[1][i] = sum[i] + dp[0][i-k];
                    loc[1][i] = i;
                }else{
                    dp[1][i] = dp[1][i-1];
                    loc[1][i] = loc[1][i-1];
                }
            }
            // three
            if(i >= 2*k){
                if(sum[i] + dp[1][i-k] > dp[2][i-1]){
                    dp[2][i] = sum[i] + dp[1][i-k];
                    loc[2][i] = i;
                }else{
                    dp[2][i] = dp[2][i-1];
                    loc[2][i] = loc[2][i-1];
                }
            }
        }
        // 回溯
        vector<int> ans(3, 0);
        int now = N-k;
        for(int i=2;i>0;i--){
            while(loc[i][now] == loc[i][now-1]) now--;
            ans[i] = loc[i][now];
            now -= k;
        }
        ans[0] = loc[0][now];
        return ans;
    }
};
// @lc code=end