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+;
 int root[maxn],a[maxn],cnt;
 vector<int>v;
 struct node{int l,r,sum;}T[maxn*];
 int getid(int x){return lower_bound(v.begin(),v.end(),x)-v.begin()+;}

 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)>>;
     if(mid>=pos)update(l,mid,T[x].l,T[y].l,pos);
     else update(mid+,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)>>;
     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+,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=;
         scanf("%d%d",&n,&m);
         FFC(i,,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,,n)update(,n,root[i],root[i-],getid(a[i]));
         FFC(i,,m)scanf("%d%d%d",&x,&y,&k),printf("%d\n",v[query(,n,root[x-],root[y],k)-]);
     }
     return ;
 }