POJ2104 K-th Number【主席树】

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN=100010;
int tot,a[MAXN],n;
int vec[MAXN],idx;
int c[MAXN*30],lson[MAXN*30],rson[MAXN*30];
int T[MAXN];
int build(int l,int r)
{
int root=tot++;
c[root]=0;
if(l!=r)
{
int mid=(l+r)>>1;
lson[root]=build(l,mid);
rson[root]=build(mid+1,r);
}
return root;
}
int update(int root,int pos,int val)
{
int newroot=tot++;
int tmp=newroot;
c[newroot]=c[root]+val;
int l=1,r=n;
while(l<r)
{
int mid=(l+r)>>1;
if(pos<=mid)
{
lson[newroot]=tot++;
rson[newroot]=rson[root];
newroot=lson[newroot];
root=lson[root];
r=mid;
}
else
{
rson[newroot]=tot++;
lson[newroot]=lson[root];
newroot=rson[newroot];
root=rson[root];
l=mid+1;
}
c[newroot]=c[root]+val;
}
return tmp;
}
int query(int l_root,int r_root,int k)
{
int l=1,r=n;
while(l<r)
{
int mid=(l+r)>>1;
if(k<=c[lson[r_root]]-c[lson[l_root]])
{
l_root=lson[l_root];
r_root=lson[r_root];
r=mid;
}
else
{
k-=c[lson[r_root]]-c[lson[l_root]];
r_root=rson[r_root];
l_root=rson[l_root];
l=mid+1;
}
}
return r;
}
int HASH(int val)
{
int l=1,r=idx;
while(l<=r)
{
int mid=(l+r)>>1;
if(vec[mid]==val)
return mid;
if(vec[mid]<val)
l=mid+1;
else
r=mid-1;
}
return l;
}
int main()
{
int m,i;
while(scanf("%d%d",&n,&m)==2)
{
for(i=1;i<=n;i++)
{
scanf("%d",&a[i]);
vec[i]=a[i];
}
tot=0;
idx=1;
for(i=2;i<=n;i++)
{
if(vec[idx]!=vec[i])
vec[++idx]=vec[i];
}
sort(vec+1,vec+idx+1);
T[0]=build(1,idx);
for(i=1;i<=n;i++)
{
int x=HASH(a[i]);
//printf("x=%d\n",x);
T[i]=update(T[i-1],x,1);
}
while(m--)
{
int l,r,x;
scanf("%d%d%d",&l,&r,&x);
printf("%d\n",vec[query(T[l-1],T[r],x)]);
}
}
return 0;
}