主席树是一个十分简单的数据结构,可以统计区间中的第k大,通过线段树的加减来实现,有n棵线段树来维护,S表示数的种数,线段树维护的就是每种数的个数
关于主席树讲解:传送门
struct Node{
Node *ch[2];int size;
Node(){ch[0]=ch[1]=NULL,size=0;}
Node(Node *ch0,Node *ch1,int s){
size=s;ch[0]=ch0,ch[1]=ch1;
}
void Updata(){
if(ch[0]) size+=ch[0]->size;if(ch[1]) size+=ch[1]->size;
}
}*root[N],*null=new Node(),q[N*9];
int cnt;int n,m,S;int a[N],a2[N];
void Build(Node *&now,Node *&last,int l,int r,int k){
if(last==NULL) last=null;
now=&q[++cnt];int m=(l+r)>>1;
if(l==r){*now=*last,now->size++;return;}
if(k<=a2[m]){
Build(now->ch[0],last->ch[0],l,m,k);
now->ch[1]=last->ch[1];now->Updata();
}
else{
Build(now->ch[1],last->ch[1],m+1,r,k);
now->ch[0]=last->ch[0];now->Updata();
}
}
void Find(Node *&L,Node *&R,int l,int r,int k){
if(L==NULL) L=null;if(R==NULL) R=null;
if(l==r){printf("%d\n",a2[l]);return;}
int m=(l+r)>>1,s=0;
if(R->ch[0]) s+=R->ch[0]->size;if(L->ch[0]) s-=L->ch[0]->size;
if(s>=k) Find(L->ch[0],R->ch[0],l,m,k);
else Find(L->ch[1],R->ch[1],m+1,r,k-s);
}
void init(){null->ch[0]=null,null->ch[1]=null;}