思路:
前缀和预处理+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