F - Load Testing CodeForces - 847H 前缀后缀和

Input

5
1 4 3 2 5

Output

6

Input

5
1 2 2 2 1

Output

1

Input

7
10 20 40 50 70 90 30

Output

0

题目大意：

问最小要加多少才能使整个数组先递增后递减（包括只递增和只递减）

解题思路：

巧妙利用前缀后缀和对每种情况（即每个点作为最高点时）进行枚举。

ac代码：

#include<bits/stdc++.h>

using namespace std;

typedef long long ll;

ll a[100005];

ll c[100005],d[100005];

ll e[100005],f[100005];

int main(){

    int n,i;
    scanf("%d",&n);
    for(i=1;i<=n;i++)
        scanf("%lld",&a[i]);
    c[1]=a[1];
    for(i=2;i<=n;i++)
    {
        c[i]=max(a[i],c[i-1]+1);//递增情况下的数组
        d[i]=d[i-1]+c[i]-a[i];//递增情况下所需加的和
        //printf("%lld %lld ",c[i],d[i]);
    }
    e[n]=a[n];
    for(i=n-1;i>=1;i--)
    {
        e[i]=max(a[i],e[i+1]+1);
        f[i]=f[i+1]+e[i]-a[i];
        //printf("%lld %lld ",e[i],f[i]);
    }
    ll minn=min(f[1],d[n]);
    ll k;
    for(i=2;i<n;i++)
    {
        k=d[i]+f[i+1]+max(c[i],e[i])-c[i];//特别注意不要忘记加上 max(c[i],e[i])-c[i]，这样才保证这个点是最高点。
        if(k<minn)
            minn=k;
    }
    printf("%lld",minn);
    return 0;
}