Codeforces Round #504 D. Array Restoration

题目：http://codeforces.com/contest/1023/problem/D

这道题目很显然，结果数组里面一定要有q，必须是连续的，他左边是递增，右边是递减的，像
1 2 2 2 3 4 3 2 1；q是4
我是实现的太烂了，老是用代码模拟脑子里的过程，用一堆if语句，将情况都分出来，还是学一学怎么实现吧，菜。。。。。
做法：如果结果中出现a[i] != 0 && a[i - 1] > a[i] && a[i] < a[i + 1]的情况，则这个情况必然不是结果数组；后面处理的都是不满足这种情况的数组，如果q出现过，那0的位置不能取q，取附近的值，如果q没出现过，那么就随便找一个有0的位置赋值为q，其他的有0的位置取附近的值。

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int inf = 0x3f3f3f3f;
const int N = 2 * 1e5 + 5;
int a[N];

int main()
{
    int n,q;
    scanf("%d %d",&n,&q);
    bool flag = false;
    for(int i = 1;i <= n;++i)
    {
        scanf("%d",&a[i]);
        if(a[i] == q){
            flag = true;
        }
    }
    for(int i = 2;i < n;++i)
    {
        if(a[i] != 0 && a[i - 1] > a[i] && a[i] < a[i + 1]){
            cout << "NO" << endl;
            return 0;
        }
    }
    int MAX = -1;
    a[0] = 1;
    for(int i = 1;i <= n;++i)
    {
        if(!flag && MAX != q && a[i] == 0){
            a[i] = q;
        }
        if(a[i] == 0){
            a[i] = a[i - 1];
        }
        MAX = max(MAX,a[i]);
    }
    if(MAX < q){
        cout << "NO" << endl;
        return 0;
    }
    cout << "YES" << endl;
    for(int i = 1;i <= n;++i)
    {
        cout << a[i] << " ";
    }
    cout << endl;
    return 0;
}