主席树POJ2104

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<vector>
#include<queue>
#include<map>
using namespace std;
struct node
{
    int id,val;
}s[100005];
int b[100005],cnt,root[100005];
bool cmp(const node &a,const node &b)
{
    return a.val<b.val;
}
struct node1
{
    int l,r,val;
}tree[100005*20];
int build(int l,int r)
{
    cnt++;
    int now=cnt;
    tree[now].val=0;
    if(l==r)
    return now;
    int mid=(l+r)>>1;
    tree[now].l=build(l,mid);
    tree[now].r=build(mid+1,r);
    return now;
}
void up(int now)
{
    tree[now].val=(tree[tree[now].l].val+tree[tree[now].r].val);
}
int update(int ori,int l,int r,int x)
{
    cnt++;
    int now=cnt;
    tree[now]=tree[ori];
    if(l==r)
    {
        tree[now].val++;
        return now;
    }
    int mid=(l+r)>>1;
    if(x<=mid)
    tree[now].l=update(tree[ori].l,l,mid,x);
    else
    tree[now].r=update(tree[ori].r,mid+1,r,x);
    up(now);
    return now;
}
int query(int l,int r,int ox,int oy,int k)
{
    if(l==r)
    return l;
    int res = tree[tree[oy].l].val- tree[tree[ox].l].val;
    int m = (l+r)/2;
    if(k <= res)
    return query(l,m,tree[ox].l,tree[oy].l,k);
    else
    return query(m+1,r,tree[ox].r,tree[oy].r,k-res);
}
int main()
{
    int n,q;
    while(cin>>n>>q)
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&s[i].val);
            s[i].id=i;
        }
        sort(s+1,s+1+n,cmp);
        for(int i=1;i<=n;i++)
        b[s[i].id]=i;
        cnt=0;
        root[0]=build(1,n);
        for(int i=1;i<=n;i++)
        {
            root[i]=update(root[i-1],1,n,b[i]);
        }
        while(q--)
        {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            int id=query(1,n,root[a-1],root[b],c);
            printf("%d\n",s[id].val);
        }
    }
    return 0;
}