CF 1037 D. Valid BFS?

D. Valid BFS?

http://codeforces.com/contest/1037/problem/D

题意：

　　给一个序列，一棵树，判断能否bfs这棵树，得到这个序列。

分析：

　　将每个点所达到的点，按照序列中出现的位置排序，然后bfs，判断是否相等。

代码：

 #include<cstdio>
 #include<algorithm>
 #include<cstring>
 #include<cmath>
 #include<iostream>
 #include<cctype>
 #include<set>
 #include<vector>
 #include<queue>
 #include<map>
 #define fi(s) freopen(s,"r",stdin);
 #define fo(s) freopen(s,"w",stdout);
 using namespace std;
 typedef long long LL;

 inline int read() {
     int x=,f=;char ch=getchar();for(;!isdigit(ch);ch=getchar())if(ch=='-')f=-;
     for(;isdigit(ch);ch=getchar())x=x*+ch-'';return x*f;
 }

 const int N = ; 

 int a[N], q[N], b[N], pos[N];
 vector<int> T[N];
 bool vis[N];

 bool cmp(int i,int j) {
     return pos[i] < pos[j];
 }

 int main() {
     int n = read();
     for (int i=; i<n; ++i) {
         int u = read(), v = read();
         T[u].push_back(v), T[v].push_back(u);
     }
     for (int i=; i<=n; ++i) a[i] = read(), pos[a[i]] = i;
     for (int i=; i<=n; ++i)
         sort(T[i].begin(), T[i].end(), cmp);
     int L = , R = , tot = ;
     q[++R] = ;
     while (L <= R) {
         int u = q[L ++];
         b[++tot] = u;
         vis[u] = true;
         for (int sz=T[u].size(),i=; i<sz; ++i) {
             int v = T[u][i];
             if (vis[v]) continue;
             q[++R] = v;
         }
     }
     for (int i=; i<=n; ++i) {
         if (a[i] != b[i]) {
             puts("No"); return ;
         }
     }
     puts("Yes");
     return ;
 }