区间k小值【模板题】

可持续化线段树

给定一个序列a1,a2,…,an，m次操作，每次给定l,r,k，问al,al+1,…,ar中第k小的值。

第一行一个正整数T(1≤T≤3)，表示测试数据的数量。
每组数据第一行n,m(1≤n,m≤100000)。
第二行n个正整数a1,a2,…,an(1≤ai≤n)。
接下来m行，每行描述一个操作，其中1≤l≤r≤n,1≤k≤r−l+1。

对于每个询问，输出一行一个整数，即第k小的值。

1
5 5
1 3 5 4 2
1 3 3
2 4 2
1 5 2
4 4 1
3 5 1

5
4
2
4
2

//左→右，即小→大
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
int n,m,tot,ver[maxn];
struct treenode{
	int l,r,sum;
}tree[maxn*20];

int build(int l,int r){
	int root=++tot;
	tree[root].sum=0;
	tree[root].l=root;
	tree[root].r=root;
	return root;
}
int change(int node,int l,int r,int pos){
	int root=++tot;
	treenode &rt=tree[root];//相当于一种指针，让新的树依旧指向旧的孩子 
	rt=tree[node],++rt.sum;
	if(l==r)return root;//到最下层了
	//寻找并改变下一层需要变动的值
	int mid=(l+r)>>1; 
	if(pos<=mid)rt.l=change(rt.l,l,mid,pos);
	else rt.r=change(rt.r,mid+1,r,pos);
	return root;
}
int query(int str,int lst,int l,int r,int k){
	if(l==r)return r;
	treenode &sn=tree[str],&ln=tree[lst];
	int temp=tree[ln.l].sum-tree[sn.l].sum;//看左边拥有数据个数
	int mid=(l+r)>>1;
	if(temp>=k)return query(sn.l,ln.l,l,mid,k);
	else return query(sn.r,ln.r,mid+1,r,k-temp);
}

int main(){
	int t,a,L,R,k;
	scanf("%d",&t);
	while(t--){
		tot=0;
		scanf("%d%d",&n,&m);
		ver[0]=build(1,n);
		//根据逐步输入的数据进行树的构建 
		for(int i=1;i<=n;i++){
			scanf("%d",&a);
			ver[i]=change(ver[i-1],1,n,a);
		}
		for(int i=1;i<=m;i++){
			scanf("%d%d%d",&L,&R,&k);
			//ver[l-1]是因为这样才能够把l的这个数据表示在这个数据区间里，从ver[l-1]到ver[r]增加的数据才算是l到r这个数据区间里面的数 
			printf("%d\n",query(ver[L-1],ver[R],1,n,k)); 
		}
	}
	return 0;
}