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;}