AtCoder Beginner Contest 336 D题 Pyramid

D题：Pyramid

标签：动态规划、前缀和
题意：金字塔型序列：$1、\mathrm{2...}k-1、k、k-\mathrm{1...2}、1$。给定一个长度为$n$的序列$a_i$，可以进行重复性的两种操作：

1. 将序列中某个数的大小减一
2. 删除第一个或最后一个数

求能够形成的金字塔型序列的最大长度。（$1<=n<=2\ast 10^5，1<=a_i<=10^9$）
题解：比较常见的套路，洛谷也有类似的题：P1091 合唱队形
从左往右维护一个$pre[i]$：以$a_i$作为结尾的最长左金字塔序列的长度
从右往左维护一个$suf[i]$：以$a_i$作为结尾的最长右金字塔序列的长度
我们以$pre[i]$为例，分别来观察一下
例子$1$：$1、2、3$
例子$2$：$1、2、2$
例子$3$：$3、1、2$
按照题目中能把数变小和删除前后数字的操作，能推出当前的$pre[i]$要从前面的$pre[i-1]$和当前$a_i$较小的那个推过来：
$pre[i]=min(pre[i-1]+1,a[i])$
$suf$同理，最终枚举每个数作为金字塔尖，左边金字塔序列长度和右边金子塔序列长度中取小的那个能够形成的金字塔序列长度，然后维护一个最大值。
代码：

#include <bits/stdc++.h>
using namespace std;

const int N = 2e5 + 10;
typedef long long ll;
ll a[N], pre[N], suf[N];

int main() {
    ll n, ans = 0;
    cin >> n;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
        pre[i] = min(pre[i - 1] + 1, a[i]);
    }
    for (int i = n; i >= 1; i--) {
        suf[i] = min(suf[i + 1] + 1, a[i]);
    }
    for (int i = 1; i <= n; i++) {
        ans = max(ans, min(pre[i], suf[i]));
    }
    cout << ans;
    return 0;
}