主席树（代码）

主席树入门

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
#define ll long long
//主席树
// 
using namespace std;
void rd() {
	#ifdef Dmymax_ly
		freopen("test.in","r",stdin);
	#endif
}
const int maxn=2e5+9;
int T,n,m,q,cnt=1,a[maxn];
vector<int>v;
int root[maxn];// 第i棵线段树的根节点下标 
struct Seg {
	int l,r; // 记录左右子树的 标号
	int sum;
} tree[maxn*20];
// 线段树 节点 

void Unique(vector<int> &v) {
	sort(v.begin(),v.end());
	v.erase(unique(v.begin(),v.end()),v.end());
}//离散化函数 

void update(int val,int &rt,int l,int r) {
	tree[cnt++]=tree[rt];// 
	rt=cnt-1;
	tree[rt].sum++;
	//能到该区间说明，这个数val一定在这个区间 
	//并且将递归到 指定的叶子结点（数组中排序后的标号） 
	if(l==r) return;//到叶子结点，退出 
	int mid=(l+r)>>1;
	if(val<=mid) update(val,tree[rt].l,l,mid);
	//权值线段树基本操作，如果这个数大于中点值，递归左子树 
	else update(val,tree[rt].r,mid+1,r);
	//否则右子树 
}
int query(int L,int R,int k,int l,int r) { //L R 查询区间  l r 递归区间
	int L_num=tree[tree[R].l].sum-tree[tree[L].l].sum;
	// 两个状态下，左子树的个数差 
	if(l==r) return v[l-1];// 递归到子节点，返回查询的值 
	int mid=(l+r)>>1;
	if(k<=L_num) return query(tree[L].l,tree[R].l,k,l,mid);
	// 左子树的值超过k ，说明第k大在 1->mid内的一个数
	// PS：(1，mid) 指的是数字大小排序 
	else return query(tree[L].r,tree[R].r,k-L_num,mid+1,r);
	// 如果左子树不够 k 个，说明第k大 在右子树
	//显然是右子树的 第 (k - L_sum)大
	//因为右子树都比左子树大  
}

int main() {
	rd();
	scanf("%d %d",&n,&q);
	for(int i=1; i<=n; i++) {
		scanf("%d",&a[i]);
		v.push_back(a[i]);
	}
	Unique(v);
	// 离散化来取标号，以及查询输出 
	int l,r,k;
	root[0]=0; tree[0].sum=tree[0].l=tree[0].r=0;
	for(int i=1; i<=n; i++) {
		k=lower_bound(v.begin(),v.end(),a[i])-v.begin()+1; 
		// 查询 a[i] 在线段树中的下标 （第几大的数） 
		root[i]=root[i-1];
		//赋值前一棵线段树的根节点 
		update(k,root[i],1,n);
		// 新一棵树的节点的根节点是在 update里改变的 
		// 所以 update 函数里 rt 参数 取地址  
	}
	for(int i=1; i<=q; i++) {
		scanf("%d %d %d",&l,&r,&k);
		printf("%d\n",query(root[l-1],root[r],k,1,n));
	}
	return 0;
}


``