poj 2104 K-th Number　(主席树)

/*
求l,r这个死序列中第k小的数
*/
#include <stdio.h>
#include <algorithm>
# include <string.h>
using namespace std;
# define lson l,m
# define rson m+1,r
# define N 100005
int a
,Hash
;
int T
;///树祖宗节点的编号
int sum[N<<5];//数目
int L[N<<5];//左儿子的编号
int R[N<<5];//右儿子的编号
int tot;
int build(int l,int r)
{
int rt=(++tot);
sum[rt]=0;
if(l<r)
{
int m=(l+r)>>1;
L[rt]=build(lson);
R[rt]=build(rson);
}
return rt;
}
int update(int pre,int l,int r,int x)
{
int rt=(++tot);
L[rt]=L[pre];
R[rt]=R[pre];
sum[rt]=sum[pre]+1;
if(l<r)
{
int m=(l+r)>>1;
if(x<=m)
L[rt]=update(L[pre],lson,x);
else
R[rt]=update(R[pre],rson,x);
}
return rt;
}
int query(int u,int v,int l,int r,int k)
{
if(l>=r)
return l;
int m=(l+r)>>1;
int num=sum[L[v]]-sum[L[u]];
if(num>=k)
return query(L[u],L[v],lson,k);
else
return query(R[u],R[v],rson,k-num);
}
int main()
{
int n,m,i;
while(~scanf("%d%d",&n,&m))
{
tot=0;
for(i=1;i<=n;i++)
{
scanf("%d",&a[i]);
Hash[i]=a[i];
}
sort(Hash+1,Hash+n+1);//离散
int size=unique(Hash+1,Hash+n+1)-Hash-1;
T[0]=build(1,size);//裸树
for(i=1;i<=n;i++)//在T[0]的基础上建n棵树
{
int x=lower_bound(Hash+1,Hash+size+1,a[i])-Hash;
T[i]=update(T[i-1],1,size,x);
}
while(m--)
{
int l,r,k;
scanf("%d%d%d",&l,&r,&k);
int x=query(T[l-1],T[r],1,size,k);
printf("%d\n",Hash[x]);
}
}
return 0;
}