poj2104 K-th number 主席树模版

这个题以前分块做，学了主席树（抄模版），存一下；

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int a[100005];
int L[100005];
struct zxs{
    int l,r,cnt;
}tree[5000005];
int tot=1;
int rt[100005];
void insert(int l,int r,int x,int &y,int v)
{
    y=++tot;
    tree[y]=tree[x];
    tree[y].cnt++;
    if(l==r) return ;
    int mid=(l+r)>>1;
    if(v<=mid)insert(l,mid,tree[x].l,tree[y].l,v);
    else    insert(mid+1,r,tree[x].r,tree[y].r,v);
}
int ask(int l,int r,int x,int y,int k)
{
    if(l==r)    return r;
    int mid=(l+r)>>1;
    int q=tree[tree[y].l].cnt-tree[tree[x].l].cnt;
    if(q>=k)return ask(l,mid,tree[x].l,tree[y].l,k);
    else    return ask(mid+1,r,tree[x].r,tree[y].r,k-q);
}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        L[i]=a[i];
    }
    sort(L+1,L+n+1);
    int len=unique(L+1,L+n+1)-L-1;
    int j;
    for(int i=1;i<=n;i++)
    {
        j=lower_bound(L+1,L+n+1,a[i])-L;
        insert(1,len,rt[i-1],rt[i],j);
    }
    int l,r,k;
    while(m--)
    {
        scanf("%d%d%d",&l,&r,&k);
        printf("%d\n",L[ask(1,len,rt[l-1],rt[r],k)]);
    }
}