CodeForces - 1037D Valid BFS?(BFS+思维)

题意

给你一棵树，再给你一个序列。

问这个序列是不是BFS序。

思路来源

http://www.cnblogs.com/zxyqzy/p/9968156.html

题解

自己对bfs序总是想当然，以为满足层序就是bfs...

但是不是，比如说1有两棵子树2和3,2有子树4,3有子树5

那么若2在3前，4必在3后5前，才是bfs序，QAQ

对动态邻接表内的点按访问id排个序就好了。

这样同一层的，就会优先满足询问序列的访问顺序，从而bfs。

代码

#include <iostream>
#include <algorithm> 
#include <cstring>
#include <cstdio>
#include <cmath>
#include <set>
#include <map>
#include <vector>
#include <stack>
#include <queue>
#include <functional>
const int INF=0x3f3f3f3f;
const int maxn=2e5+10; 
const int mod=1e9+7;
const int MOD=998244353;
const double eps=1e-7;
typedef long long ll;
#define vi vector<int> 
#define si set<int>
#define pii pair<int,int> 
#define pi acos(-1.0)
#define pb push_back
#define mp make_pair
#define lowbit(x) (x&(-x))
#define sci(x) scanf("%d",&(x))
#define scll(x) scanf("%lld",&(x))
#define sclf(x) scanf("%lf",&(x))
#define pri(x) printf("%d",(x))
#define rep(i,j,k) for(int i=j;i<=k;++i)
#define per(i,j,k) for(int i=j;i>=k;--i)
#define mem(a,b) memset(a,b,sizeof(a)) 
using namespace std;
int n,id[maxn],a[maxn];
vector<int>G[maxn],ans;
queue<int>q;
bool vis[maxn];
void add(int x,int y)
{
	G[x].push_back(y);
}
bool cmp(int x,int y)
{
	return id[x]<id[y];
}
void bfs(int s)
{
	q.push(s);
	while(!q.empty())
	{
		int tmp=q.front(),len=G[tmp].size();
		q.pop();
		if(vis[tmp])continue;
		vis[tmp]=1;
		ans.push_back(tmp);
		rep(i,0,len-1)
		if(!vis[G[tmp][i]])
		q.push(G[tmp][i]);
	}
}
int main()
{ 
   sci(n);
   rep(i,0,n-2)
   {
   	int x,y;
   	sci(x),sci(y);
   	x--,y--;
   	add(x,y);
   	add(y,x);
   }
   rep(i,0,n-1)
   {
   	int x;
   	sci(a[i]);
	a[i]--;
   	id[a[i]]=i;
   }
   rep(i,0,n-1)sort(G[i].begin(),G[i].end(),cmp);
   bfs(0);
   bool flag=1;
   rep(i,0,n-1)
   {
   	if(ans[i]-a[i])
   	{
   		flag=0;
   		break;
   	}
   }
   if(!flag)puts("No");
   else puts("Yes"); 
   return 0;
}