【NOIP2015模拟11.3】装饰大楼

Description

给出一个序列A，Ai表示对于一个h序列来讲，以i结尾的lis的长度。h中的数两两不等。
现在你知道了A删去一个数之后的序列B（未知删掉哪位），求A序列有多少种。
|A|<=10^6

Solution

好多细节呀！！！
不爽，懒得打题解。
可以发现，对于一个序列A，它满足条件的前提就是，对于每一个Ai，都存在一个Aj(j< i)满足Aj+1=Ai。（dp大家都会打吧）
那么，我们枚举插入的位置，这个位置Ai的取值范围就是1~mex(Aj(j< i))+1。
判重判无解就行了。
不爽，懒得打。

Code

#include
#include
#include
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
#define N 1000005
#define ll long long
using namespace std;
int a[N],bz[N],n,k;
ll ans;
int main() {
    freopen("building.in","r",stdin);
    freopen("building.out","w",stdout);
    scanf("%d",&n);k=n;bz[0]=1;
    fo(i,1,n-1) {
        scanf("%d",&a[i]);
        if (!bz[a[i]]) {
            if (!bz[a[i]-1]) {
                if (a[i]>1&&!bz[a[i]-2]||kprintf("0");return 0;}
                k=min(k,i);
            }
        }bz[a[i]]=1; 
    }
    if (kint l=0;
        fo(i,0,k-1) {
            l=max(l,a[i]);if (l+1>=a[k]-1) ans++;
        }
        printf("%lld",ans);return 0;
    }k=0;a[n]=0x7fffffff;
    fo(i,1,n-1) {
        k=max(k,a[i]);if (k+1>=a[i+1]) ans--;
        ans+=(ll)k+1;
    }
    printf("%lld",ans);
}