HDU 2665 kth number(主席树)

//
//  main.cpp
//  Richard
//
//  Created by 邵金杰 on 16/9/23.
//  Copyright © 2016年 邵金杰. All rights reserved.
//

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=100000+10;
struct node{
int l,r,sum;
}tree[maxn*40];
int n,m;
int root[maxn],a[maxn];
vector<int> v;
int tot;
int get_id(int x) {return (int)(lower_bound(v.begin(),v.end(),x)-v.begin()+1);}
void update(int l,int r,int &x,int y,int pos)
{
tree[++tot]=tree[y];
tree[tot].sum++;
x=tot;
if(l==r) return ;
int mid=(l+r)>>1;
if(pos<=mid) update(l,mid,tree[x].l,tree[y].l,pos);
else         update(mid+1,r,tree[x].r,tree[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=tree[tree[y].l].sum-tree[tree[x].l].sum;
if(k<=sum) return query(l,mid,tree[x].l,tree[y].l,k);
else       return query(mid+1,r,tree[x].r,tree[y].r,k-sum);
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&m);
tot=0;
v.clear();
memset(root,0,sizeof(root));
for(int i=1;i<=n;i++) {scanf("%d",&a[i]);v.push_back(a[i]);}
sort(v.begin(),v.end());
v.erase(unique(v.begin(),v.end()),v.end());
for(int i=1;i<=n;i++) update(1,(int)v.size(),root[i],root[i-1],get_id(a[i]));
for(int i=0;i<m;i++){
int x,y,k;
scanf("%d%d%d",&x,&y,&k);
printf("%d\n",v[query(1,(int)v.size(),root[x-1],root[y],k)-1]);
}
}
return 0;
}