poj2104 主席树裸题

#include <cstdio>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn=1e5+5;
struct Node
{
     int L,R,num;
}tree[20*maxn];
struct Rank
{
     int x,id;
     bool operator <(Rank a)const
     {
         return x<a.x;
     }
}ran[maxn];
int root[maxn],num[maxn];
int cnt=1;
void updata(int x,int &rt,int a,int b)
{
    tree[cnt++]=tree[rt];
    rt=cnt-1;
    tree[rt].num++;
    if(a==b)return;
    int mid=(a+b)/2;
    if(x<=mid)
        updata(x,tree[rt].L,a,mid);
    else
        updata(x,tree[rt].R,mid+1,b);
}
int quer(int a,int b,int k,int s,int o)
{
    int d=tree[tree[b].L].num-tree[tree[a].L].num;
    if(s==o)return s;
    if(k<=d)
        return quer(tree[a].L,tree[b].L,k,s,(s+o)/2);
    else
        return quer(tree[a].R,tree[b].R,k-d,(s+o)/2+1,o);
}
int main()
{
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&ran[i].x);
        ran[i].id=i;
    }
    sort(ran+1,ran+n+1);
    for(int i=1;i<=n;i++)
    {
        num[ran[i].id]=i;
    }
    for(int i=1;i<=n;i++)
    {
        root[i]=root[i-1];
        updata(num[i],root[i],1,n);
    }
    for(int i=1;i<=m;i++)
    {
        int l,r,k;
        scanf("%d %d %d",&l,&r,&k);
        printf("%d\n",ran[quer(root[l-1],root[r],k,1,n)].x);
    }
    return 0;
}