题解:刚开始把所有点离点1 的距离求出来,并且纪录每个距离的点数,然后逆序依次判断每一个点,是否符合当前距离,如果枚举完了当前距离的点数,就让距离减一,同距离的点判断依据是父节点的出现顺序,我们再找一下就可以了
#include <bits/stdc++.h>
#define inf 0x3f3f3f3f
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;
const int N=2e5+10;
int ar[N],du[N],vis[N],op[N],pre[N],pos[N];
vector<int>G[N];
int n,mx;
void dfs(int x,int cnt)
{
vis[x]=1;
du[x]=cnt;
op[cnt]++;
mx=max(mx,cnt);
for(int i=0;i<G[x].size();i++){
int y=G[x][i];
if(!vis[y]){
pre[y]=x;
dfs(y,cnt+1);
}
}
}
int main()
{
#ifdef local
freopen("D://rush.txt","r",stdin);
#endif
cin>>n;
mx=1;
for(int i=0;i<n-1;i++)
{
int a,b;
cin>>a>>b;
G[a].push_back(b),G[b].push_back(a);
}
for(int i=0;i<n;i++)
cin>>ar[i],pos[ar[i]]=i;
dfs(1,1); pre[1]=1;
memset(vis,0,sizeof vis);
int cnt=mx,last=-1 ;
for(int i=n-1;i>=0;i--){
if(du[ar[i]]!=cnt||vis[ar[i]]){cout<<"NO"<<endl;return 0;}
op[cnt]--,vis[ar[i]]=1;
int a=pos[pre[ar[i]]];
if(last==-1) last=a;
else if(last<a) {cout<<"NO"<<endl;return 0;}
else last=a ;
if(op[cnt]==0) cnt--;
}
cout<<"YES"<<endl;
return 0;
}