主席树 模板

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include<queue>
#include<set>
#include<stack>
#include<string>
#include<sstream>
#include<vector>
#define inf 2147483647
#define mem0(a) memset(a,0,sizeof(a))
#define rep(i,a,b) for(int i=a;i<b;i++)
#define  rr read()
#define ll long long
#define pn putchar('\n')
#define ps putchar(' ')
#define repb(i,a,b) for(int i=a;i<=b;i++)
#define CloseIo   ios_base::sync_with_stdio(false), cin.tie(0), cout.tie(0)
using namespace std;
const int N=2e5+5;
int node_cnt=0;//给结点分配的数值
int sum[N<<5],rt[N],lc[N<<5],rc[N<<5];

//sum表示该区间有几个数,rt表示每颗线段树的根结点的坐标，lc和rc分别表示左子树和右子树的位置
int a[N],b[N];//原序列和离散序列
int p;//修改点，与p相关都是要改变的那条链，需要设立新结点
void build(int &t,int left,int right) {
	t=++node_cnt;//将地址传递过来，赋值(即每个区间代表的结点值，rt,而sum初始都为0)
	if(left==right)return;
	int mid=(left+right)>>1;
	build(lc[t],left,mid);
	build(rc[t],mid+1,right);
}
int modify(int p_root,int left,int right) { //加入一个数就相当于加到第几个数的树 就改变从根到叶子（该点为第几小）路径上的sum都加一
	int root=++node_cnt;//同样的，给每一个结点赋值
	lc[root]=lc[p_root];
	rc[root]=rc[p_root];
	sum[root]=sum[p_root]+1;//直接先连接原来的左右子树 然后该区间的值加一
	//一直走到叶子节点的路径上的值都加一
	if(left==right)return root;
	int mid=(left+right)>>1;
	if(p<=mid)lc[root]=modify(lc[root],left,mid);//如果左子树涉及P,设立新结点，改变左子树的结点值
	else rc[root]=modify(rc[root],mid+1,right);//如果右子树涉及P,设立新结点，改变右子树的结点值
	return root;//记得返回该区间的结点值，该结点是父节点的左结点或右结点，确定结点值
}
int query(int u,int v,int left,int right,int k) {
	int ans,x=sum[lc[v]]-sum[lc[u]];//x表示，[left,right]区间，左子树（左区间）有几个数
	if(left==right)return left;
	int mid=(left+right)>>1;
	if(x>=k)ans=query(lc[u],lc[v],left,mid,k);//如果左边的数大于等于K个,说明第K小的数在左子树（或者左区间）
	else
		ans=query(rc[u],rc[v],mid+1,right,k-x);//反之，就在右区间的第k-x个小的数
	return ans;
}
int main() {
	int n,m;
// freopen("testdata.in","r",stdin);
// freopen("res.txt","w",stdout);
	int T;
	scanf("%d",&T);
	while(T--) {
        node_cnt=0;
        rt[0]=0;
		scanf("%d %d",&n,&m);
		for(int i=1; i<=n; i++) {
			scanf("%d",&a[i]);
			b[i]=a[i];
		}
		sort(b+1,b+1+n);
		int q=unique(b+1,b+1+n)-b-1;//q表示不重复的数有几个，也表示主席树的最大右区间，[1,q]
		//区间[x,y]的值表示大于等于x，小于等于y的数有几个，这个用sum[k](k，结点值，代表这个区间)数组来记录
		//那么,结合下面的p和上面的q来看，这段程序其实将N个元素离散成1~q，然后再插入1~q插入到主席树中

		build(rt[0],1,q);//从0开始建 初始状态就是有0个数
		for(int i=1; i<=n; i++) {
			p=lower_bound(b+1,b+1+q,a[i])-b;
//这里p得到的是什么呢，它是a[i]这个元素在序列中是第几大(也就是在b中的下标)的数
			rt[i]=modify(rt[i-1],1,q);//每插入第i个数就生成第i个状态的树 第i棵树的访问方式由该函数返回根节点
			//每次都是在前一个状态的树上建立新树
		}
		while(m--) {
			int l,r,k;
			scanf("%d %d %d",&l,&r,&k);
			int ans=query(rt[l-1],rt[r],1,q,k);//得到的ans表示在b数组里面第ans小的数
			printf("%d\n",b[ans]);//最后输出b[ans]就是答案
		}
	}
}