主席树水题 HDU2665

和POJ 2104一模一样，不多说

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5+6;
struct node {int l,r,sum;}T[maxn*20];
int root[maxn],a[maxn],cnt;
vector <int> q;
int getid(int x){return lower_bound(q.begin(),q.end(),x)-q.begin()+1;};
void update(int l,int r,int x,int &y,int pos)
{
    T[++cnt]=T[x];T[cnt].sum++;y=cnt;
    if(l==r) return ;
    int m = (l+r)>>1;
    if(m>=pos) update(l,m,T[x].l,T[y].l,pos);
    else update(m+1,r,T[x].r,T[y].r,pos);
}
int query(int l,int r,int x,int y,int k)
{
    if(l==r) return l;
    int sum=T[T[y].l].sum-T[T[x].l].sum;
    int m = (l+r)>>1;
    if(sum>=k) query(l,m,T[x].l,T[y].l,k);
    else query(m+1,r,T[x].r,T[y].r,k-sum);
}
int main()
{
    int i,n,m,t,x,y,k;
    scanf("%d",&t);
    while(t--)
    {
        cnt=0;q.clear();
        scanf("%d%d",&n,&m);
        for(i=1;i<=n;i++) scanf("%d",a+i),q.push_back(a[i]);
        sort(q.begin(),q.end());q.erase(unique(q.begin(),q.end()),q.end());
        for(i=1;i<=n;i++) update(1,n,root[i-1],root[i],getid(a[i]));
        for(i=1;i<=m;i++)
        {
            scanf("%d%d%d",&x,&y,&k);
            printf("%d\n",q[query(1,n,root[x-1],root[y],k)-1]);
        }
    }
    return 0;
}