POJ 2104 K-th 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+100;
int n,m;
struct node{
    int l,r,sum;
}tree[maxn*40];
vector<int> v;
int get_pos(int x) {return (int)(lower_bound(v.begin(),v.end(),x)-v.begin()+1);}
int a[maxn],root[maxn];
int tot;
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 sum=tree[tree[y].l].sum-tree[tree[x].l].sum;
    int mid=(l+r)>>1;
    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()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        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_pos(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;
}