主席树求区间第K小（大）数模板

#include <cstdio>
#include <algorithm>

using namespace std;

const int MAXN = 1e5+10;

struct T{
	int L,R;//分别指向左右子树
	int sum;//该节点所管辖区间范围内数的个数
	T(){
		sum = 0;
	}
}Tree[MAXN*20];//尽量开大点，否则容易RE甚至T。

struct V{
	int value,id;//值的大小，离散之前在原数组中的位置 
	bool operator < (const struct V &b)const{
		return value < b.value;//按降序排结果就是第K大 
	}
}board[MAXN];

int root[MAXN];//多颗线段树的根节点
int Rank[MAXN];//原数组离散之后的数组
int tot;

void Updata(int num,int &rt,int left,int right){
	Tree[++tot] = Tree[rt];
	rt = tot;
	Tree[rt].sum++;
	if(left == right)return ;
	int mid = left + (right-left)/2;
	if(num <= mid)Updata(num,Tree[rt].L,left,mid);
	else Updata(num,Tree[rt].R,mid+1,right);
}

int Query(int ql,int qr,int k,int left,int right){
	int t = Tree[Tree[qr].L].sum - Tree[Tree[ql].L].sum;
	if(left == right)return left;
	int mid = left + (right-left)/2;
	if(k <= t)return Query(Tree[ql].L,Tree[qr].L,k,left,mid);
	else return Query(Tree[ql].R,Tree[qr].R,k-t,mid+1,right);
}

inline void init(){
	tot = 0;
	root[0] = 0;
	Tree[0].L = Tree[0].R = Tree[0].sum = 0;
}

int main(){
	
	int T;//Case数 
	scanf("%d",&T);
	int N,M;//N是数的个数，M是查询的数量。 
	while(T--){
		scanf("%d %d",&N,&M);
		init();
		for(int i=1 ; i<=N ; ++i){
			scanf("%d",&board[i].value);
			board[i].id = i;
		}
		//-----------------------------------------进行离散化
		sort(board+1,board+1+N);
		for(int i=1 ; i<=N ; ++i)Rank[board[i].id] = i;
		//--------------------------------------------------- 
		for(int i=1 ; i<=N ; ++i){
			root[i] = root[i-1];
			Updata(Rank[i],root[i],1,N);
		}
		int left,right,k;//查询区间和第几小（大） 
		for(int i=1 ; i<=M ; ++i){
			scanf("%d %d %d",&left,&right,&k);
			printf("%d
",board[Query(root[left-1],root[right],k,1,N)].value);
		}
	}
	
	return 0;
}

PS：这里模板中的查询区间是 [left,right] 即闭区间。