HDU 2665 Kth number（主席树）

套模板。。同poj 2104。莫名其妙的re的两发。。。

#include <bits/stdc++.h>
using namespace std;

const int MAXN = 1e6+10;
int num[MAXN],root[MAXN];
struct node
{
    int l,r,sum;
}T[MAXN*40];
int cnt,n,m;
vector<int> vec;

int getid(int x)
{
    return lower_bound(vec.begin(),vec.end(),x) - vec.begin() + 1;
}

void init()
{
    vec.clear();
    cnt = 0;
    root[0] = 0;
    T[0].l = T[0].r = T[0].sum = 0;
}

void update(int l, int r, int &x, int y, int pos)
{
    T[++cnt] = T[y];
    T[cnt].sum++;
    x = cnt;
    if(l == r) return;
    int mid = (l+r) >> 1;
    if(pos <= mid) update(l,mid,T[x].l,T[y].l,pos);
    else update(mid+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 mid = (l+r) >> 1;
    int sum = T[T[y].l].sum - T[T[x].l].sum;
    if(k <= sum) return query(l,mid,T[x].l,T[y].l,k);
    else return query(mid+1,r,T[x].r,T[y].r,k-sum);
}

int main()
{
    int T,x,y,z;
    scanf("%d",&T);
    while(T--)
    {
        init();
        scanf("%d %d",&n,&m);
        for(int i = 1; i <= n; ++i)
        {
            scanf("%d",&num[i]);
            vec.push_back(num[i]);
        }
        sort(vec.begin(),vec.end());
        vec.erase(unique(vec.begin(),vec.end()),vec.end());
        for(int i = 1; i <= n; ++i)
            update(1,n,root[i],root[i-1],getid(num[i]));
        for(int i = 1; i <= m; ++i)
        {
            scanf("%d %d %d",&x,&y,&z);
            printf("%d\n",vec[query(1,n,root[x-1],root[y],z)-1]);
        }
    }
    return 0;
}