Feed the dogs POJ - 2761 (主席树第K大)

题意：

找l~r第k大的数

#include <iostream>
#include <stdio.h>
#include <map>
#include <algorithm>
using namespace std;
const int maxn=100001;
int a[maxn];
int root[maxn*20];
struct node
{
    int lc,rc,num;
}tree[maxn*20];
int cnt,rnu,now,number[maxn];
void build(int &i,int l,int r)
{
    tree[++cnt]=tree[i];
    i=cnt;
    if(l==r)
        return ;
    int mid=(l+r)>>1;
    build(tree[i].lc,l,mid);
    build(tree[i].rc,mid+1,r);
}
void update(int &i,int l,int r,int num)
{
    tree[++cnt]=tree[i];
    i=cnt;
    tree[i].num++;
    if(l==r)
        return ;
    int mid=(l+r)>>1;
    if(num<=a[mid])
        update(tree[i].lc,l,mid,num);
    else
        update(tree[i].rc,mid+1,r,num);
}
int query(int i,int j,int l,int r,int k)
{
    if(l==r)
        return a[l];
    int mid=(l+r)>>1;
    if(tree[tree[j].lc].num-tree[tree[i].lc].num>=k)
        query(tree[i].lc,tree[j].lc,l,mid,k);
    else
        query(tree[i].rc,tree[j].rc,mid+1,r,k-(tree[tree[j].lc].num-tree[tree[i].lc].num));
}
int main()
{

    int n,m;
    scanf("%d%d",&n,&m);
    cnt=rnu=now=0;
    for(int i=1;i<=n;i++)
        scanf("%d",&number[i]),a[i]=number[i];
    sort(a+1,a+1+n);
    rnu=unique(a+1,a+1+n)-a-1;
   // cout<<rnu<<endl;
    build(root[0],1,rnu);
    for(int i=1;i<=n;i++)
    {
        now++;
        update(root[now]=root[now-1],1,rnu,number[i]);
    }
    for(int i=1;i<=m;i++)
    {
        int l,r,k;
        scanf("%d%d%d",&l,&r,&k);
        printf("%d\n",query(root[l-1],root[r],1,rnu,k));
    }
}