#include<cstdio>
#include<algorithm>
#include<cstring>
#include<ctime>
#include<iostream>
using namespace std;
#define rep(i,x,y) for(int i=x;i<=y;++i)
#define dep(i,x,y) for(int i=x;i>=y;--i)
struct Treap
{Treap *l,*r;
int fix,key,size;
Treap(int val):fix(rand()),key(val),l(NULL),r(NULL),size(1){}//这什么鬼写法woc...
inline void updata(){size=1+(l?l->size:0)+(r?r->size:0);}
} *root=NULL;
typedef pair<Treap*,Treap*> Droot;
inline int Size(Treap *x){return (x?x->size:0);}
Treap *Merge(Treap *A,Treap *B)//本人以fix较小的放在顶端
{if(!A) return B;
if(!B) return A;
if(A->fix<B->fix)
{A->r=Merge(A->r,B);A->updata();return A;}
else
{B->l=Merge(A,B->l);B->updata();return B;}
}
Droot Split(Treap *x,int k)//这精髓部分。。。
{ if(!x) return Droot(NULL,NULL);
Droot y;
if(Size(x->l)>=k)//第k小的在左子树中
{y=Split(x->l,k);//1~k,k+1~n;
x->l=y.second;
x->updata();
y.second=x;
}else{
y=Split(x->r,k-Size(x->l)-1);
x->r=y.first;
x->updata();
y.first=x;
}
return y;
}
int Findkth(int k){//查找第K小
Droot x=Split(root,k-1);
Droot y=Split(x.second,1);
Treap *ans=y.first;
root=Merge(Merge(x.first,ans),y.second);
return ans->key;
}
int Getkth(Treap *x,int v){//询问一个数是第几小
if(!x)return 0;
return v<x->key?Getkth(x->l,v):Getkth(x->r,v)+Size(x->l)+1;//注意要写v<x->key
}
void Insert(int v){//插入操作
int k=Getkth(root,v);
Droot x=Split(root,k);
Treap *n=new Treap(v);
root=Merge(Merge(x.first,n),x.second);//插入一定要按照从小到大的顺序
}
void Delete(int k){//删除操作
Droot x=Split(root,k-1);//将1~k-1放在x.first里,k~n放在x.second里
Droot y=Split(x.second,1);//将k放在y.first里,k+1~n放在y.second里
root=Merge(x.first,y.second);//于是k就被抛弃了。。。
}
int a[100005],M,x,y;
int main(){
scanf("%d",&a);
rep(i,1,a[0])
{ scanf("%d",a+i);Insert(a[i]);}
scanf("%d",&M);
while(M--){
char ch=getchar();
while(ch!='Q' && ch!='A' && ch!='D') ch=getchar();
scanf("%d",&x);
if(ch=='Q') printf("%d\n",Findkth(x));
if(ch=='A') Insert(x);
if(ch=='D') Delete(x);
}
return 0;
}
可持久化treap 模版
最新推荐文章于 2022-11-12 22:38:56 发布