Manthan, Codefest 18 (rated, Div. 1 + Div. 2) D. Valid BFS?（思维）

题目链接


题意：给定一棵树，现在规定对于一棵bfs树的遍历过程中，兄弟节点的入队顺序可以任意变换，给出一个序列问是不是bfs遍历序列的一种？
思路：我们知道bfs是广搜也就是一层一层来的，所以dfs的时候可以记录深度，如果同层次的点深度不同肯定no，除了深度还有一个点要注意，如果同深度的节点，i先入队，再入队j，那么轮到i和j的子树的时候也一定是i的子树先入队，j的子树随后，那么这个怎么记录呢？我们只要记录每个点的前驱，只要前驱的位置关系和子树的位置关系一样就行了。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e5+5;
vector<int>g[maxn],temp,ans,v;
int a[maxn],dep[maxn],id[maxn],pre[maxn];
void dfs(int u,int fa,int deep)
{
	dep[u]=deep;
	pre[u]=fa;
	for(int to:g[u])
	{
		if(to==fa) continue;
		dfs(to,u,deep+1);
	}
}
int main()
{
	int n,t;
	scanf("%d",&n);
	for(int i=1,u,v;i<n;++i)
	{
		scanf("%d%d",&u,&v);
		g[u].push_back(v);
		g[v].push_back(u);
	}
	dfs(1,0,1);
	for(int i=1;i<=n;++i) scanf("%d",&t),v.push_back(t),temp.push_back(dep[t]),ans.push_back(dep[t]),id[t]=i;
	sort(temp.begin(),temp.end());
	if(temp!=ans){
		puts("No");return 0;
	}
	for(int i=1;i<v.size();++i)
	{
		if(dep[v[i]]!=dep[v[i-1]]) continue;
		else{
			if(pre[v[i]]==pre[v[i-1]]) continue;
			else{
				if(id[pre[v[i]]]<id[pre[v[i-1]]]){
					puts("No");return 0;
				}
			}
		}
	}
	puts("Yes");
}