codeforces 1291B Array Sharpening 贪心

https://vjudge.net/problem/CodeForces-1291B
题目大意：给一个数组，$0<=a_i<=10^9$，你可以对任意$a_i>0$使得$a_i=a_i-1$。问经过若干次操作后，能否找到一个$k(1<=k<=n)$，使得$a_1<\dots \dots <a_k$且$a_k>\dots \dots >a_n$。

思路：不妨先假设$k>1$，也就是说$a_1,\dots \dots ,a_k$一定是单调上升的，那么根据贪心策略，该序列一定可以转换成最优序列：$0,1,\dots \dots ,k-1$。也就是说如果$a[i]>=i$，则可以令$a[i]=i$，如果$a[i]<i$，说明出现了拐点,也就是说$a_{i-1},a_i,\dots \dots ,a_n$必须要单调下降，所以有$a_i=min(a_i,a_{i-1}-1)$，如果出现了$a_i<0$则说明不符题意。那么$k=1$的情况怎么处理呢？我们可以用一个数组$b$，$b$是$a$的逆序表示，然后再用上述方法对$b$进行判断，如果可行就是满足题意的。

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;

const int maxn=3e5+5;

int n;
int a[2][maxn];

bool check(int id)
{
    int idx=-1;
    for(int i=0;i<n;i++)
    {
        if(a[id][i]>=i)
            a[id][i]=i;
        else
        {
            idx=i;
            break;
        }
    }
    if(idx==-1)
        return 1;
    bool flag=1;
    for(int i=idx;i<n&&flag;i++)
    {
        if(a[id][i]>=a[id][i-1])
            a[id][i]=a[id][i-1]-1;
        if(a[id][i]<0)
            flag=0;
    }
    return flag;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        for(int i=0;i<n;i++)
            scanf("%d",&a[0][i]);
        for(int i=0;i<n;i++)
            a[1][i]=a[0][n-i-1];
        if(check(0)||check(1))
            printf("YES\n");
        else
            printf("NO\n");
    }
    return 0;
}