HDU 2665 Kth number [可持久化线段树 主席树]

题意：给n个数的数组，m次询问，每次询问区间【L,R】的第k小

题解：裸的主席树，练练手

AC代码：

#include<stdio.h>
#include<algorithm>
#include<map> 
using namespace std;
#define N 100005
int root[N];
int size[N*25],lchild[N*25],rchild[N*25];
int tot;
void insert(int last,int cur,int L,int R,int k)
{
	size[cur]=size[last];
	if(L==R)
	{
		size[cur]++;
		return ;
	}
	int mid=L+R>>1;
	if(k<=mid)
	{
		insert(lchild[last],lchild[cur]=++tot,L,mid,k);
		rchild[cur]=rchild[last];
	}
	else
	{
		insert(rchild[last],rchild[cur]=++tot,mid+1,R,k);
		lchild[cur]=lchild[last];
	}
	size[cur]=size[lchild[cur]]+size[rchild[cur]];
}
int query(int last,int cur,int k,int L,int R)
{
	if(L==R)
		return L;
	int mid=L+R>>1;
	int nows=size[lchild[cur]]-size[lchild[last]];
	if(nows>=k)return query(lchild[last],lchild[cur],k,L,mid);
	else return query(rchild[last],rchild[cur],k-nows,mid+1,R);
}
int a[N],b[N];
map<int,int>mp;
int main()
{
	int T;
	scanf("%d",&T);
	while(T--)
	{
		tot=0;
		int ss=0,n,m;
		scanf("%d%d",&n,&m);
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			b[i]=a[i]; 
		}
		sort(b+1,b+n+1);
		int t=unique(b+1,b+n+1)-b;
		t--;
		for(int i=1;i<=t;i++)mp[b[i]]=++ss;
		for(int i=1;i<=n;i++)
		{
			root[i]=++tot;
			insert(root[i-1],root[i],1,t,mp[a[i]]);
		}
		for(int i=0;i<m;i++)
		{
			int l,r,k;
			scanf("%d%d%d",&l,&r,&k);
			printf("%d\n",b[query(root[l-1],root[r],k,1,t)]);
		}
	}
}