主席数基础模板

不带修改主席树 

#include<bits/stdc++.h>
#define il inline
#define pb push_back
#define fi first
#define se second
#define ms(_data,v) memset(_data,v,sizeof(_data))
#define sc(n) scanf("%d",&n)
#define SC(n,m) scanf("%d %d",&n,&m)
#define SZ(a) int((a).size())
#define rep(i,a,b) for(int i=a;i<=b;++i)
#define drep(i,a,b)	for(int i=a;i>=b;--i)
using namespace std;
typedef long long ll;
const ll inf=0x3f3f3f3f;
const double PI=acos(-1.0);
const double eps=1e-9;
const int maxn=1e5+5;
int a[maxn],b[maxn],rt[maxn*20],ls[maxn*20],rs[maxn*20],sum[maxn*20];
int n,m,tot;
void build(int &rt,int l,int r){
	rt=++tot;
	sum[rt]=0;
	if(l==r) return;
	int mid=(l+r)>>1;
	build(ls[rt],l,mid);
	build(rs[rt],mid+1,r); 
}
void update(int &rt,int l,int r,int last,int p){
	rt=++tot;
	ls[rt]=ls[last],rs[rt]=rs[last];
	sum[rt]=sum[last]+1;
	if(l==r) return;
	int mid=(l+r)>>1;
	if(p<=mid) update(ls[rt],l,mid,ls[last],p);
	else update(rs[rt],mid+1,r,rs[last],p);
}
int query(int ss,int ee,int l,int r,int k){
	if(l==r) return l;
	int mid=(l+r)>>1;
	int cnt=sum[ls[ee]]-sum[ls[ss]];
	if(k<=cnt) return query(ls[ss],ls[ee],l,mid,k);
	else return query(rs[ss],rs[ee],mid+1,r,k-cnt);
}
int main(){
	SC(n,m);
	rep(i,1,n) sc(a[i]),b[i]=a[i];
	sort(b+1,b+n+1);
	int sz=unique(b+1,b+n+1)-(b+1);
	build(rt[0],1,sz);
	rep(i,1,n){
		a[i]=lower_bound(b+1,b+sz+1,a[i])-b; 
		update(rt[i],1,sz,rt[i-1],a[i]);	
	} 
	int l,r,k;
	rep(i,1,m){
		sc(l),sc(r),sc(k);
		int id=query(rt[l-1],rt[r],1,sz,k);
		printf("%d\n",b[id]);
	} 
	return 0;
}

带修改的主席数

#include<bits/stdc++.h>
#define il inline
#define pb push_back
#define sc(n) scanf("%d",&n)
#define SC(n,m) scanf("%d %d",&n,&m)
#define rep(i,a,b) for(int i=a;i<=b;++i)
#define ms(_data,v) memset(_data,v,sizeof(_data))
#define SZ(a) int((a).size())
using namespace std;
typedef long long ll;
const ll inf=0x3f3f3f3f;
const int maxn=1e5+5;
const int maxm=maxn*150;
int n,m,tot,cnt,sz,a[maxn],b[maxn*5];
int root[maxn],ls[maxm],rs[maxm],dat[maxm];
int totL,totR;
int L[maxn],R[maxn];
template <typename _Tp> il void read(_Tp&x) {
	char ch;bool flag=0;x=0;
	while(ch=getchar(),!isdigit(ch)) if(ch=='-')flag=1;
	while(isdigit(ch)) x=x*10+ch-'0',ch=getchar();
	if(flag) x=-x;
}
il void insert(int &now, int l, int r, int x, int val){
    if(!now) now=++tot;
    dat[now]+=val;
    if(l==r) return;
    int mid=(l+r)>>1;
    if(x<=mid) insert(ls[now],l,mid,x,val);
    else insert(rs[now],mid+1,r,x,val);
}
il int ask(int l, int r, int k){
	if(l==r) return l;
	int sum=0;
    for(int i=1; i<=totL; i++) sum-=dat[ls[L[i]]];
    for(int i=1; i<=totR; i++) sum+=dat[ls[R[i]]];
	int mid=(l+r)>>1;
	if(sum>=k){
		for(int i=1; i<=totL; i++) L[i]=ls[L[i]];
        for(int i=1; i<=totR; i++) R[i]=ls[R[i]];
        return ask(l,mid,k);
	}
	else{
		for(int i=1; i<=totL; i++) L[i]=rs[L[i]];
        for(int i=1; i<=totR; i++) R[i]=rs[R[i]];
		return ask(mid+1,r,k-sum);
	}
}
il int lowbit(int x){return x&-x;}
il void update(int x,int C){
    int t=a[x],tt=C;
    a[x]=C;
    for(int j=x; j<=n; j+=lowbit(j)){
        insert(root[j],1,sz,t,-1);
        insert(root[j],1,sz,tt,1);
    }
}
il int query(int l,int r,int t){ //返回下标 
    totL=totR=0,l--;
    for(int j=l; j; j-=lowbit(j)) L[++totL]=root[j];
    for(int j=r; j; j-=lowbit(j)) R[++totR]=root[j];
    int id=ask(1,sz,t);
    return id; 
}
struct node{
	int op,l,r,k;
}q[maxn];
int main(){
	read(n),read(m);
	tot=0,cnt=0;
	rep(i,1,n) sc(a[i]),b[++cnt]=a[i];
	char op;
	int l,r,k;
	rep(i,1,m){
		op=getchar();
		while(!(op=='Q' || op=='C')) op=getchar();
		if(op=='Q'){
			read(l),read(r),read(k);
			q[i]=node{1,l,r,k};
		}
		else{
			read(l),read(r);
			b[++cnt]=r;
			q[i]=node{0,l,r,0};
		}
	}
	sort(b+1,b+cnt+1);
	sz=unique(b+1,b+cnt+1)-(b+1);
	rep(i,1,n){
		a[i]=lower_bound(b+1,b+sz+1,a[i])-(b);
		for(int j=i;j<=n;j+=lowbit(j)){
			insert(root[j],1,sz,a[i],1);
		}
	}
	rep(i,1,m){
		if(q[i].op){ //query
			int id=query(q[i].l,q[i].r,q[i].k);
			printf("%d\n",b[id]);
		}
		else{ // modify
			int x=q[i].l,now=lower_bound(b+1,b+sz+1,q[i].r)-b;
			update(x,now);
		}
	}
	return 0;
}