【提醒自己】CodeForces - 847H Load Testing

Problem Description

给你一个长度为n的数组，让你把它变成递增，或者递减，或者先增后减。问你最少需要加多少数

思路：

一开始写烂了，还以为自己的代码很对。后面发现没考虑好。
做法:
求出递增需要的最少，递减需要的最少。
然后再跑一边for求，先增后减的最少

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define maxn 100055

ll a[maxn], lb[maxn], rb[maxn];
ll lsum[maxn], rsum[maxn];
int main()
{
    int n, i;
    while(~scanf("%d", &n))
    {
        for(i = 0; i < n; i++)
            scanf("%lld", &a[i]);
        memcpy(lb, a, sizeof(a));
        memset(lsum, 0, sizeof(lsum));
        memset(rsum, 0, sizeof(rsum));
        for(i = 1; i < n; i++)//递增
        {
            if(lb[i] <= lb[i-1])
            {
                lsum[i] = lsum[i-1] + lb[i-1] - lb[i] + 1;
                lb[i] = lb[i-1] + 1;//记录前缀和
            }
            else lsum[i] = lsum[i-1];
        }
        memcpy(rb, a, sizeof(a));
        for(i = n - 1; i >= 1; i--)//递减
        {
            if(rb[i-1] <= rb[i])
            {
                rsum[i-1] = rsum[i] + rb[i] - rb[i-1] + 1;
                rb[i-1] = rb[i] + 1;//记录后缀和
            }
            else rsum[i-1] = rsum[i];
        }
        ll ans = min(lsum[n-1], rsum[0]);
       // printf("%lld\n", ans);
        for(i = 1; i < n - 1; i++)
        {
            //例如4 4 5 不能变成4 5 5, 得变成4 6 5
            ll c1 = (lb[i] == rb[i+1])?1:0;//核心点，容易漏掉的地方
            ll c2 = (rb[i] == lb[i-1])?1:0;
            ans = min(ans, lsum[i] + rsum[i+1] + c1);
            ans = min(ans, lsum[i-1] + rsum[i] + c2);
        }
        printf("%lld\n", ans);
    }
}