cf - 1037D - Valid BFS? - ( bfs + 思维 ）

题目链接：点击进入

题意

给一颗树，给一个序列，问你这个序列是不是这棵树的合法的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;
}