【Leetcode】689. Maximum Sum of 3 Non-Overlapping Subarrays

题目地址：

https://leetcode.com/problems/maximum-sum-of-3-non-overlapping-subarrays/description/

给定一个长$n$数组$a$，题目保证$\forall i,a[i]>0$。再给定一个正整数$k$，题目保证$k\le \lfloor \frac{n}{3}\rfloor$。现在要求在$a$中找到$3$个长$k$的不重合的区间，使得这$3$个区间的总和最大，问方案，只需给出每个区间的起点下标即可。如果方案不唯一，则返回字典序最小的那个方案。

思路是动态规划。设$a$的下标从$1$开始，先求前缀和数组$s$。设$f[i][j]$是从$a$的第$i$个数开始的后缀中取$j$个不重合子区间的情况下的最大总和（递推从后向前递推，那么求方案的时候就能从前向后求字典序最小的方案）。那么全局最大总和就是$f[1][3]$。如果不选$a_i$，则最大总和为$f[i+1][j]$；如果选$a_i$，那么$a_i$所在区间的总和为$s[i+k-1]-s[i-1]$，总最大总和为$f[i+k][j-1]+s[i+k-1]-s[i-1]$。所以 f [ i ] [ j ] = max ⁡ { f [ i + 1 ] [ j ] , f [ i + k ] [ j − 1 ] + s [ i + k − 1 ] − s [ i − 1 ] } f[i][j]=\max\{f[i+1][j],f[i+k][j-1]+s[i+k-1]-s[i-1]\} f[i][j]=max{f[i+1][j],f[i+k][j−1]+s[i+k−1]−s[i−1]}最后从状态$(x,y)=(1,3)$开始递推，逐个验证$a_x$是否能取。代码如下：

class Solution {
 public:
  vector<int> maxSumOfThreeSubarrays(vector<int>& a, int k) {
    int n = a.size();
    vector<int> s(n + 1);
    for (int i = 1; i <= n; i++) s[i] = s[i - 1] + a[i - 1];
    vector<vector<int>> f(n + 2, vector<int>(4));
    for (int i = n - k + 1; i; i--)
      for (int j = 1; j <= 3; j++)
        f[i][j] = max(f[i + 1][j], f[i + k][j - 1] + s[i + k - 1] - s[i - 1]);

    vector<int> res;
    for (int x = 1, y = 3; y;) {
      while (f[x][y] != f[x + k][y - 1] + s[x + k - 1] - s[x - 1]) x++;
      res.push_back(x - 1);
      x += k, y--;
    }
    return res;
  }
};

时空复杂度$O(n)$。