描述
初始有一个空集,依次插入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;
}