「BZOJ1803」Spoj1487 Query on a tree III

题目大意：给一棵n个节点的树，每个点有一个权值$val_i$，m次查询，每次查询x的子树中第k大的节点的编号，权值不重复，且树以1为根。

做法：看起来就是一个查询区间第k大强行上树，于是dfs序做

#include<bits/stdc++.h>
using namespace std;
#define ll long long
inline ll read() {
    ll x=0,f=1; char ch=getchar();
    for(;ch<'0'||ch>'9';ch=getchar())
        if(ch=='-')f=-f;
    for(;ch>='0'&&ch<='9';ch=getchar())
        x=x*10+ch-'0';
    return x*f;
}
inline void chkmin( int &a,int b ) { if(a>b) a=b; }
inline void chkmax( int &a,int b ) { if(a<b) a=b; }
#define _ read()
#define ln endl
const int N=1e5+5;
vector < int > G[N];
int n,m,a[N],val[N],b[N],tot,id[N];
int T[N],sum[20*N],l[N],r[N],lc[20*N],rc[20*N];
inline void insert( int v,int l,int r,int x,int &y ) {
    y=++tot; sum[y]=sum[x]+1;
    if(l==r) return;
    int mid=(l+r)>>1;
    lc[y]=lc[x]; rc[y]=rc[x];
    if(v<=mid) insert(v,l,mid,lc[x],lc[y]); 
    else insert(v,mid+1,r,rc[x],rc[y]);
}
inline int query( int k,int l,int r,int x,int y ) {
    if(k>sum[y]-sum[x]) return 0;
    if(l==r) return l;
    int mid=(l+r) >> 1;
    if(sum[lc[y]]-sum[lc[x]]>=k) return query(k,l,mid,lc[x],lc[y]);
    else return query(k-(sum[lc[y]]-sum[lc[x]]),mid+1,r,rc[x],rc[y]);
}
inline void dfs( int x,int fa ) {
    l[x]=++tot; val[tot]=a[x];
    for( int i=0;i<G[x].size();i++ ) {
        int y=G[x][i];
        if(y!=fa) dfs(y,x);
    }
    r[x]=tot;
}
int main() {
    n=_;
    for( int i=1;i<=n;i++ ) a[i]=_,b[i]=a[i];
    sort(b+1,b+n+1); m=unique(b+1,b+n+1)-b-1;
    for( int i=1;i<=n;i++ ) a[i]=lower_bound(b+1,b+m+1,a[i])-b;
    for( int i=1;i<=n;i++ ) b[a[i]]=i;
    for( int i=1;i<n;i++ ) {
        int x=_,y=_;
        G[x].push_back(y); G[y].push_back(x);
    } dfs(1,0);
    for( int i=1;i<=n;i++ ) insert(val[i],1,n,T[i-1],T[i]);
    m=_;
    while(m--) { 
        int x=_,k=_;
        printf("%d\n",b[query(k,1,n,T[l[x]-1],T[r[x]])]);
    }
}