暑假集训 第N天 T2 Number

问题 B: Number
时间限制: 1 Sec 内存限制: 256 MB
提交: 44 解决: 36
[提交][状态][讨论版]
题目描述
一个排列，求出了 a 数组，其中 ai 表示第 i 个数左边有多少个数比它小

计算出原来的排列

输入
第一行输入 n 接下来 n - 1 个整数 ai，下标从 2 开始

输出
输出 n 个整数表示原排列

样例输入
5
1
2
1
0
样例输出
2
4
5
3
1
提示

对于 20% 的数据满足：1 ≤ n ≤ 10


对于 50% 的数据满足：1 ≤ n ≤ 1000

对于 100% 的数据满足，1 ≤ n ≤ 100000

第一眼，我貌似只会      给数列求左边有多少比自己小的    （怎么恰巧反过来了，好尴尬）= =

第二眼，逆着推了一遍样例，貌似有感觉了，哈哈，无旋treap秒之

当我们计算一个数左边有多少比其小时，右边的数完全没有用，甚至还会产生干扰（在不知道具体大小前）

所以我们不如倒着来。

给出了比自己小的个数，就等于知道了其排名（明白了吗？（排名！！！））

注意  由于treap是有序的（我们一定维护升序啊）

所以所以在查第i个数排名时是必须将[i+1,N]的数删去，（又验证了要倒着来）

即在查第i个数排名时，treap中必须只有前i个数，即查完一个数就将其删去。

看样例，最后一个是0，即在前5个数中第5个数排名1，又因为数为[1，N]，所以第5个数就是1，然后将其删去。

              第4个是1，即在剩下5个数中第4个数排名第二，为3  （1已经删去），记录后再将其删去。

......以此类推即可

最后带一波无旋treap    无旋treap大法吼！！！！！

附上代码：

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
using namespace std;
const int M=100000+10;
int n,pre[M],num[M],hj;
struct treap{
    treap* ch[2];
    int size,val,key;
    inline treap(int v)  {size=1;val=v;key=rand();ch[0]=ch[1]=NULL;}
    inline void update() {size=1+(ch[0]?ch[0]->size:0)+(ch[1]?ch[1]->size:0);}
}*rt;
typedef pair<treap*,treap*> dou;
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9') {if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') {x=(x<<1)+(x<<3)+(ch^48);ch=getchar();}
    return x*f;
}
inline int getsize(treap* o){return o?o->size:0;}
inline treap* merge(treap* a,treap* b){
    if(!a) return b;
    if(!b) return a;
    if(a->key<b->key) {a->ch[1]=merge(a->ch[1],b);a->update();return a;}
    else              {b->ch[0]=merge(a,b->ch[0]);b->update();return b;}
}
inline dou split(treap* o,int k){
    if(!o) return dou(NULL,NULL);
    if(getsize(o->ch[0])>=k){
        dou x=split(o->ch[0],k);
        o->ch[0]=x.second;o->update();
        x.second=o;return x;
    }else{
        dou x=split(o->ch[1],k-getsize(o->ch[0])-1);
        o->ch[1]=x.first;o->update();
        x.first=o;return x;
    }
}
inline int getrank(treap* o,int v){
    if(!o) return 0;
    if(o->val>=v) return getrank(o->ch[0],v);
    else {return getrank(o->ch[1],v)+getsize(o->ch[0])+1;}
}
inline int findrank(int k){
    dou x=split(rt,k-1);
    dou y=split(x.second,1);
    treap* ans=y.first;
    rt=merge(x.first,merge(y.first,y.second));
    return ans?ans->val:0;
}
inline void _insert(int v){
    int k=getrank(rt,v);
    dou x=split(rt,k);
    treap* o=new treap(v);
    rt=merge(merge(x.first,o),x.second);
}
inline void _delete(int v){
    int k=getrank(rt,v);
    dou x=split(rt,k);
    dou y=split(x.second,1);
    rt=merge(x.first,y.second);
}
int main(){
    n=read();
    for(int i=1;i<=n;i++) _insert(i);
    for(int i=2;i<=n;i++) pre[i]=read();
    for(int i=n;i>=2;i--){
        int v=findrank(pre[i]+1);
        _delete(v);
        num[i]=v;
    }
    num[1]=rt->val;
    for(int i=1;i<=n;i++)
        printf("%d\n",num[i]);
    return 0;
}