【Leetcode】2355. Maximum Number of Books You Can Take

题目地址：

https://leetcode.com/problems/maximum-number-of-books-you-can-take/description/

给定一个长$n$非负整数数组$A$，可以构造一个下标从$i$开始的严格递增的数列$a_i,a_{i+1},...$满足$0\le a_k\le A_k$，求所有满足这样条件的数列的最大总和。

设$f[i]$为以下标$i$结尾的和最大的严格递增数列的总和。设这个数列是$a_k,a_{k+1},...,a_i$，容易知道一定有$a_i=A_i$，我们尽量让$a_j=A_i-(i-j),j<i$才能使得总和最大。我们找到$A_i$左边最近的$A_j$满足$A_j<A_i-(i-j)$，那么从$i$向左走到$j+1$的时候，最大和可以累加上 A i , A i − 1 , . . . , max ⁡ { 0 , A i − ( j − i ) } A_i,A_i-1,...,\max\{0,A_i-(j-i)\} Ai​,Ai​−1,...,max{0,Ai​−(j−i)}，再加上$f[j]$（因为这个最大的尽量$a_j=A_i-(i-j),j<i$的数列向左延伸到$j$的时候延伸不动了）；当然如果满足条件的$A_j$不存在是最好的，$A_i,A_i-1,...$可以一路延续到左边直到值变成$0$或者下标变成$0$。

问题就转变为求每个$A_i$的左边最近的满足$A_j<A_i-(i-j)$的$j$。这可以用单调栈来做，因为如果$A_j\ge A_i-(i-j)$的话，对于更大的$i$肯定更满足，从而可以将其下标pop掉。

代码如下：

class Solution {
 public:
  using LL = long long;
  LL maximumBooks(vector<int>& v) {
    int n = v.size();
    stack<int> stk;
    long res = 0;
    vector<long> f(n);
    for (int i = 0; i < n; i++) {
      while (stk.size() && v[stk.top()] > v[i] - (i - stk.top())) stk.pop();
      if (stk.size()) {
        int len = i - stk.top();
        f[i] = f[stk.top()] + (long)(v[i] + v[i] - len + 1) * len / 2;
      } else {
        // 这里说明可以延伸到最左边，那么两种可能，一种是值变为0，另一种下标变为0
        int len = v[i] - i <= 0 ? v[i] : i + 1;
        f[i] = (long)(v[i] + v[i] - len + 1) * len / 2;
      }
      res = max(res, f[i]);
      stk.push(i);
    }
    return res;
  }
};

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