bzoj 1803: Spoj1487 Query on a tree III(主席树)

题意

你被给定一棵带点权的n个点的有根数，点从1到n编号。

定义查询 query(x,k): 寻找以x为根的k大点的编号（从小到大排序第k个点）

假设没有两个相同的点权。

输入格式： 第一行为整数n，第二行为点权，接下来n-1行为树边，接下来一行为整数m，下面m行为两个整数x,k，代表query(x,k)

输出格式： m行，输出每次查询的结果。

题解

先一遍dfs，然后建个主席树，带上去直接跑一跑就好了

我忘了注意dfs序的位置和原来的编号……结果调了半天啥都调不出来……

 //minamoto
 #include<iostream>
 #include<cstdio>
 #include<algorithm>
 using namespace std;
 #define getc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
 char buf[<<],*p1=buf,*p2=buf;
 inline int read(){
     #define num ch-'0'
     char ch;bool flag=;int res;
     while(!isdigit(ch=getc()))
     (ch=='-')&&(flag=true);
     for(res=num;isdigit(ch=getc());res=res*+num);
     (flag)&&(res=-res);
     #undef num
     return res;
 }
 char obuf[<<],*o=obuf;
 inline void print(int x){
     if(x>) print(x/);
     *o++=x%+;
 }
 const int N=,M=N*;
 int sum[M],L[M],R[M],rt[N];
 int ver[N<<],Next[N<<],head[N];
 int ls[N],rs[N],a[N],b[N],id[N],pos[N];
 int n,m,cnt,tot,q;
 void update(int last,int &now,int l,int r,int x){
     sum[now=++cnt]=sum[last]+;
     if(l==r) return;
     int mid=(l+r)>>;
     if(x<=mid) R[now]=R[last],update(L[last],L[now],l,mid,x);
     else L[now]=L[last],update(R[last],R[now],mid+,r,x);
 }
 int query(int u,int v,int l,int r,int k){
     if(l>=r) return l;
     int x=sum[L[v]]-sum[L[u]];
     int mid=(l+r)>>;
     if(x>=k) return query(L[u],L[v],l,mid,k);
     else return query(R[u],R[v],mid+,r,k-x);
 }
 inline void add(int u,int v){
     ver[++tot]=v,Next[tot]=head[u],head[u]=tot;
     ver[++tot]=u,Next[tot]=head[v],head[v]=tot;
 }
 void dfs(int u,int fa){
     a[ls[u]=++m]=b[u],id[m]=u;
     for(int i=head[u];i;i=Next[i])
     if(ver[i]!=fa) dfs(ver[i],u);
     rs[u]=m;
 }
 int main(){
     //freopen("testdata.in","r",stdin);
     n=read();
     for(int i=;i<=n;++i) b[i]=read();
     for(int i=;i<n;++i){
         int u,v;
         u=read(),v=read();
         add(u,v);
     }
     dfs(,);
     sort(b+,b++m);
     for(int i=;i<=n;++i){
         int k=lower_bound(b+,b++m,a[i])-b;
         update(rt[i-],rt[i],,m,k);
         pos[k]=id[i];
     }
     q=read();
     while(q--){
         int u=read(),k=read();
         int ans=pos[query(rt[ls[u]-],rt[rs[u]],,m,k)];
         print(ans),*o++='\n';
     }
     fwrite(obuf,o-obuf,,stdout);
     return ;
 }