2018.08.27 rollcall(非旋treap)

描述

初始有一个空集,依次插入N个数Ai。有M次询问Bj,表示询问第Bj个数加入集合后的排名为j的数是多少

输入

第一行是两个整数N,M

接下来一行有N个整数,Ai

接下来一行有M个整数Bj,保证数据合法

输出

M行,回答每个询问

样例输入

7 4

9 7 2 8 14 1 8

1 2 6 6

样例输出

9

9

7

8

提示

【说明】

第一次询问,当前集合{9},1th=9

第二次询问,当前集合{9,7}的第2=9

第三次询问,当前集合{9,7,2,8,14,1}的第3=7

第四次询问,当前集合{9,7,2,8,14,1}的第4=8

【数据规模】

40%的数据保证 n ≤ 1000

100%的数据保证1≤m≤n≤30000;0≤Ai<2^32

康复训练ing。。。

一道裸的不能再裸的非旋treap的模板题。。。

代码:

#include<bits/stdc++.h>
#define ll long long
#define N 30005
using namespace std;
inline ll read(){
    ll ans=0;
    char ch=getchar();
    while(!isdigit(ch))ch=getchar();
    while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
    return ans;
}
inline void write(ll x){
    if(x>9)write(x/10);
    putchar((x%10)^48);
}
ll a[N],ans[N];
struct Node{int tim,id;}b[N];
int n,m;
typedef pair<int,int> pii;
struct fhq_treap{
    int rt,cnt,son[N][2],rd[N],siz[N];
    ll val[N];
    inline int build(ll v){siz[++cnt]=1,rd[cnt]=rand(),val[cnt]=v,son[cnt][0]=son[cnt][1]=0;return cnt;}
    inline void pushup(int p){siz[p]=siz[son[p][0]]+siz[son[p][1]]+1;}
    inline int merge(int a,int b){
        if(!a||!b)return a+b;
        if(rd[a]>rd[b]){son[a][1]=merge(son[a][1],b),pushup(a);return a;}
        son[b][0]=merge(a,son[b][0]),pushup(b);return b;
    }
    inline pii split(int p,int k){
        if(!p)return pii(0,0);
        pii tmp;
        if(siz[son[p][0]]>=k){
            tmp=split(son[p][0],k);
            son[p][0]=tmp.second,pushup(p);
            return make_pair(tmp.first,p);
        }
        tmp=split(son[p][1],k-siz[son[p][0]]-1);
        son[p][1]=tmp.first,pushup(p);
        return make_pair(p,tmp.second);
    }
    inline int rank(int p,ll v){
        if(!p)return 0;
        if(val[p]>v)return rank(son[p][0],v);
        return siz[son[p][0]]+1+rank(son[p][1],v);
    }
    inline ll kth(int k){
        pii x=split(rt,k),y=split(x.first,k-1);
        rt=merge(merge(y.first,y.second),x.second);
        return val[y.second];
    }
    inline void insert(ll v){
        int k=rank(rt,v),p=build(v);
        pii x=split(rt,k);
        rt=merge(merge(x.first,p),x.second);
    }
}T;
inline bool cmp(Node a,Node b){return a.tim<b.tim;}
int main(){
    srand(time(NULL));
    n=read(),m=read();
    for(int i=1;i<=n;++i)a[i]=read();
    for(int i=1;i<=m;++i)b[i].tim=read(),b[i].id=i;
    sort(b+1,b+m+1,cmp);
    for(int i=1;i<=m;++i){
        Node q=b[i];
        for(int j=b[i-1].tim+1;j<=q.tim;++j)T.insert(a[j]);
        ans[q.id]=T.kth(q.id);
    }
    for(int i=1;i<=m;++i)write(ans[i]),puts("");
    return 0;
}
上一篇:2018.08.05 bzoj3223: Tyvj 1729 文艺平衡树(非旋treap)


下一篇:python特殊函数 __call__()