COJ1013 WZJ的数据结构（十三）

#include<cstdio>
#include<cctype>
#include<cstring>
#include<algorithm>
using namespace std;
inline int read()
{
char ch=getchar();int sig=1,x=0;
while(!isdigit(ch)) {if(ch=='-') sig=-1;ch=getchar();}
while(isdigit(ch)) x=x*10+ch-'0',ch=getchar();
return x*sig;
}
const int maxn=100010;
const int maxnode=20000010;
int s[maxnode],ls[maxnode],rs[maxnode],ToT;
void update(int& y,int x,int l,int r,int pos)
{
s[y=++ToT]=s[x]+1;if(l==r) return;
ls[y]=ls[x];rs[y]=rs[x];int mid=l+r>>1;
if(pos<=mid) update(ls[y],ls[x],l,mid,pos);
else update(rs[y],rs[x],mid+1,r,pos);
}
int query(int x,int y,int l,int r,int k)
{
if(l==r) return l;
int mid=l+r>>1;
if(s[ls[y]]-s[ls[x]]>=k) return query(ls[x],ls[y],l,mid,k);
return query(rs[x],rs[y],mid+1,r,k-s[ls[y]]+s[ls[x]]);
}
int to[maxn*2],next[maxn*2],first[maxn],e;
void AddEdge(int a,int b)
{
to[++e]=b;next[e]=first[a];first[a]=e;
to[++e]=a;next[e]=first[b];first[b]=e;
}
int L[maxn],R[maxn],v[maxn],root[maxn],pos[maxn],ts,siz[maxn];
void dfs(int x,int fa)
{
L[x]=++ts;pos[ts]=x;siz[x]=1;
for(int i=first[x];i;i=next[i]) if(fa!=to[i]) dfs(to[i],x),siz[x]+=siz[to[i]];
R[x]=ts;
}
int main()
{
int n=read(),q=read(),x,k;
for(int i=1;i<n;i++) AddEdge(read(),read());
for(int i=1;i<=n;i++) v[i]=read();
dfs(1,0);
for(int i=1;i<=n;i++) update(root[i],root[i-1],1,n,v[pos[i]]);
while(q--)
{
x=read();k=read();
if(siz[x]>=k) printf("%d\n",query(root[L[x]-1],root[R[x]],1,n,k));
else puts("-1");
}
return 0;
}