【JZOJ A组】简单的填数

Description

Input

第一行一个数 n，表示序列的长度。
第二行 n 个整数，第 i 个整数表示 ai，如果 ai = 0，则表示这个位置没有填数。

Output

如果不存在合法的填数方案，则输出 −1； 否则第一行输出一个整数，表示最大的 an；第二行 n 个正整数，第 i 个数表示完 成填数后的序列的第 i 个元素。 如果有多组合法的解，输出任意一组

Sample Input

【样例 1 输入】
7
0 1 0 0 0 3 0
【样例 2 输入】
4
0 0 0 3

Sample Output

【样例 1 输出】
3
1 1 2 2 3 3 3
【样例 2 输出】
-1

Data Constraint

对于 30% 的数据，n ≤ 1000；
对于另外 30% 的数据，数据保证随机生成；
对于 100% 的数据，2 ≤ n ≤ 2 × 10^5 , 0 ≤ ai ≤ 10^5。

思路

我们可以先定义一个up数组和一个down数组，up表示可能走到的最大值，down表示可能走到的最小值

那么对于没有填的数，先不考虑填了的数，up两个进1，down五个进1肯定是它们要表示的值
那么如果对于填了的数，先考虑上界
那么现将上界定到填的数，然后长度为2，如果填的数等于上界，在上界求出来的长度和2取min
下界的话，将下界定到填的数，长度为1
如果最后up[n].l=1的话，也就说明了up[n].x不可以填的，不是最大值
因为在所有情况都是最优下，只剩1的长度，没有比这种情况更优的，所以up[n].x是填不了的
现在最大值也确定了，只要找到一种方法填数就好了
然后，从后往前跑，那么对于有填的数，直接s[a[i]]++
没填的数，在a[i+1]和up[i].x取min值
如果s[min]=5的话，往前走一位就好了
还有-1的情况，有以下几种情况：
①a[i]>1
②up[n].x

代码

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=100077;
struct node
{
    int x,l;
}up[maxn*2],down[maxn*2];
int n,a[maxn*2],s[maxn];
int main()
{
    scanf("%d",&n);
    for(int i=1; i<=n; i++) scanf("%d",&a[i]);
    if(a[1]>1)
    {
        printf("-1"); return 0;
    }
    up[1].x=1; up[1].l=1;
    down[1].x=1; down[1].l=1;
    for(int i=2; i<=n; i++)
    {
        down[i]=down[i-1]; up[i]=up[i-1];
        if(++up[i].l>2) up[i].x++,up[i].l=1;
        if(++down[i].l>5) down[i].x++,down[i].l=1;
        if(a[i])
        {
            if(up[i].x>a[i]) up[i].x=a[i],up[i].l=2; else if(up[i].x==a[i]) up[i].l=min(up[i].l,2);
            if(down[i].x<a[i]) down[i].x=a[i],down[i].l=1;
            if(up[i].x<a[i]||down[i].x>a[i]) 
            {
                printf("-1"); return 0;
            }
        }
    }
    if(up[n].l==1) up[n].x--,up[n].l=up[n-1].l+1;
    if(up[n].x<down[n].x)
    {
        printf("-1"); return 0;
    }
    printf("%d\n",a[n]=up[n].x);
    s[a[n]]=1;
    for(int i=n-1; i>=1; i--)
    {
        if(!a[i])
        {
            int t=min(a[i+1],up[i].x);
            if(s[t]==5) t--;
            a[i]=t;
        }
        s[a[i]]++;
    }
    for(int i=1; i<=n; i++) printf("%d ",a[i]);
}