题目链接:点击进入
题意
给一颗树,给一个序列,问你这个序列是不是这棵树的合法的bfs序列。
思路
bfs遍历树,每次取一个点,标记它的子节点,再从序列中接着取出与它的子节点的数量相同的点,看是否有之前出现过的点,或者有子节点中未出现的点,有就不合法,没有就则继续向后遍历。
代码
//#pragma GCC optimize(2)
//#pragma GCC optimize(3)
#include<iostream>
#include<string>
#include<map>
#include<set>
//#include<unordered_map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#include<iomanip>
#include<cmath>
#include<fstream>
#define X first
#define Y second
#define best 131
#define INF 0x3f3f3f3f
#define pii pair<int,int>
#define lowbit(x) x & -x
//#define int long long
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double eps=1e-7;
const double pai=acos(-1.0);
const int N=2e4+10;
const int maxn=1e6+10;
const int mod=1e9+7;
int head[maxn],tot,n,a[maxn];
bool vis[maxn],vv[maxn];
struct node
{
int to;
int next;
}edge[maxn];
void init()
{
memset(head,-1,sizeof(head));
memset(vis,0,sizeof(vis));
memset(vv,0,sizeof(vv));
tot=0;
}
void add(int u,int v)
{
edge[tot].to=v;
edge[tot].next=head[u];
head[u]=tot++;
}
bool bfs()
{
if(a[1]!=1) return 0;
int pos=2;queue<int>q;
q.push(1);vis[1]=1;
while(q.size())
{
int now=q.front(),cnt=0;
q.pop();
for(int i=head[now];i!=-1;i=edge[i].next)
if(!vis[edge[i].to])
cnt++,vv[edge[i].to]=1;
for(int i=pos;i<=pos+cnt-1;i++)
{
if(vv[a[i]]==0||vis[a[i]]) return 0;
vis[a[i]]=1,q.push(a[i]);
}
pos+=cnt;
for(int i=head[now];i!=-1;i=edge[i].next)
vv[edge[i].to]=0;
}
return 1;
}
int main( )
{
ios::sync_with_stdio(false);
init();
cin>>n;
for(int i=1;i<n;i++)
{
int x,y;
cin>>x>>y;
add(x,y);
add(y,x);
}
for(int i=1;i<=n;i++) cin>>a[i];
if(bfs()) cout<<"Yes"<<endl;
else cout<<"No"<<endl;
return 0;
}