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;
}