【题目链接】

           ​​点击打开链接​

【算法】

           dfs序 + 线段树

           树链剖分同样可以解决这个问题

【代码】

            



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

int i,n,p,a,b,timer;
int tag[MAXN<<2],sum[MAXN<<2],size[MAXN],fa[MAXN],dfn[MAXN];

vector< int > e[MAXN];

inline void dfs(int u)
{
int i,v;
dfn[u] = ++timer;
size[u] = 1;
for (i = 0; i < e[u].size(); i++)
{
v = e[u][i];
if (fa[u] != v)
{
fa[v] = u;
dfs(v);
size[u] += size[v];
}
}
}
inline void pushdown(int index)
{
if (tag[index])
{
sum[index<<1] += tag[index];
sum[index<<1|1] += tag[index];
tag[index<<1] += tag[index];
tag[index<<1|1] += tag[index];
tag[index] = 0;
}
}
inline void update(int index)
{
sum[index] = sum[index<<1] + sum[index<<1|1];
}
inline void modify(int index,int l,int r,int ql,int qr,int val)
{
int mid;
if (l == ql && r == qr)
{
sum[index] += val;
tag[index] += val;
return;
}
pushdown(index);
mid = (l + r) >> 1;
if (mid >= qr) modify(index<<1,l,mid,ql,qr,val);
else if (mid + 1 <= ql) modify(index<<1|1,mid+1,r,ql,qr,val);
else
{
modify(index<<1,l,mid,ql,mid,val);
modify(index<<1|1,mid+1,r,mid+1,qr,val);
}
update(index);
}
inline int query(int index,int l,int r,int pos)
{
int mid;
if (l == r) return sum[index];
pushdown(index);
mid = (l + r) >> 1;
if (mid >= pos) return query(index<<1,l,mid,pos);
else return query(index<<1|1,mid+1,r,pos);
}

int main() {

scanf("%d",&n);
for (i = 1; i < n; i++)
{
scanf("%d%d",&a,&b);
e[a].push_back(b);
e[b].push_back(a);
}
dfs(1);
for (i = 1; i <= n; i++)
{
scanf("%d",&p);
printf("%d\n",query(1,1,n,dfn[p]));
modify(1,1,n,dfn[p],dfn[p]+size[p]-1,1);
}

return 0;

}