学习笔记（数据结构）— 主席树

主席树

用途：查询区间第k大
核心思想：前缀和，权值线段树

过程

（1）对于每一个点都建一颗权值线段树,表示到当前节点各个数出现了几次
（2）每次查询两棵线段树相减，类似前缀和，得出当前区间内各个数出现的次数，再利用线段树的结构进行类似二分的查找即可

优化

对每个点都建树显然容易 memory limit exceeded
思考发现，一个点只会对上一个点的权值线段树中每一层的一个点的值产生影响，所以可以将其他节点的root直接指向上一颗树的对应的节点，这样增加的空间大概是 $logn$ 级别的，乘以线段树 $nlogn$ 的空间，最后的空间复杂度大概是在 $O(n\ast logn\ast logn)$ 级别（一般开32倍）

代码

#include <bits/stdc++.h>
#define ll long long
#define endl '\n'
using namespace std;
const int maxn=2e5+5;
int a[maxn],b[maxn],sz=0,p;
int sum[maxn<<5],lc[maxn<<5],rc[maxn<<5],rt[maxn<<5];
void build(int &rt,int l,int r){
	rt=++sz;sum[rt]=0;
	if(l==r) return ;
	int mid=(l+r)>>1;
	build(lc[rt],l,mid);build(rc[rt],mid+1,r);
} 
int update(int id,int l,int r){
	int idd=++sz;
	rc[idd]=rc[id];lc[idd]=lc[id];sum[idd]=sum[id]+1;
	if(l==r) return idd;
	int mid=(l+r)>>1;
	if(mid>=p) lc[idd]=update(lc[idd],l,mid);
	else rc[idd]=update(rc[idd],mid+1,r);
	return idd;
}
int query(int u,int v,int l,int r,int k){
	int x=sum[lc[v]]-sum[lc[u]],mid=(l+r)>>1;
	if(l==r) return l;
	if(x>=k) return query(lc[u],lc[v],l,mid,k);
	else return query(rc[u],rc[v],mid+1,r,k-x);
}
int main(){
	//ios::sync_with_stdio(false);cin.tie(0);
	int n,T;cin>>n>>T;
	for(int i=1;i<=n;i++) cin>>a[i],b[i]=a[i];
	sort(b+1,b+n+1);
	int q=unique(b+1,b+n+1)-b-1;
	build(rt[0],1,q);
	for(int i=1;i<=n;i++){
		p=lower_bound(b+1,b+q+1,a[i])-b;
		rt[i]=update(rt[i-1],1,q);
	}
	while(T--){
		int l,r,k;
		cin>>l>>r>>k;
		cout<<b[query(rt[l-1],rt[r],1,q,k)]<<endl;
	}
}