hdu_2665_Kth number(主席树)

题目连接：http://acm.hdu.edu.cn/showproblem.php?pid=2665

题意：给你一个区间，让你找这个区间第K大的数

题解：主席树模版题，也可以用划分树

#include<cstdio>
#include<vector>
#include<algorithm>
#define FFC(i,a,b) for(int i=a;i<=b;++i)
using namespace std;

const int maxn=1e5+7;
int root[maxn],a[maxn],cnt;
vector<int>v;
struct node{int l,r,sum;}T[maxn*40];
int getid(int x){return lower_bound(v.begin(),v.end(),x)-v.begin()+1;}

void update(int l,int r,int &x,int y,int pos){
	T[++cnt]=T[y],T[cnt].sum++,x=cnt;
	if(l==r)return;
	int mid=(l+r)>>1;
	if(mid>=pos)update(l,mid,T[x].l,T[y].l,pos);
	else update(mid+1,r,T[x].r,T[y].r,pos);
}

int query(int l,int r,int x,int y,int k){
	if(l==r)return l;
	int mid=(l+r)>>1;
	int sum=T[T[y].l].sum-T[T[x].l].sum;
	if(sum>=k)return query(l,mid,T[x].l,T[y].l,k);
	else return query(mid+1,r,T[x].r,T[y].r,k-sum);
}
int main(){
	int t,m,n,x,y,k;
	scanf("%d",&t);
	while(t--){
		v.clear();cnt=0;
		scanf("%d%d",&n,&m);
		FFC(i,1,n)scanf("%d",&a[i]),v.push_back(a[i]);
		sort(v.begin(),v.end()),v.erase(unique(v.begin(),v.end()),v.end());
		FFC(i,1,n)update(1,n,root[i],root[i-1],getid(a[i]));
		FFC(i,1,m)scanf("%d%d%d",&x,&y,&k),printf("%d\n",v[query(1,n,root[x-1],root[y],k)-1]);
	}
	return 0;
}