CF 1037 D. Valid BFS?
D. Valid BFS?
http://codeforces.com/contest/1037/problem/D
题意:
给一个序列,一棵树,判断能否bfs这棵树,得到这个序列。
分析:
将每个点所达到的点,按照序列中出现的位置排序,然后bfs,判断是否相等。
代码:
1 #include<cstdio> 2 #include<algorithm> 3 #include<cstring> 4 #include<cmath> 5 #include<iostream> 6 #include<cctype> 7 #include<set> 8 #include<vector> 9 #include<queue> 10 #include<map> 11 #define fi(s) freopen(s,"r",stdin); 12 #define fo(s) freopen(s,"w",stdout); 13 using namespace std; 14 typedef long long LL; 15 16 inline int read() { 17 int x=0,f=1;char ch=getchar();for(;!isdigit(ch);ch=getchar())if(ch=='-')f=-1; 18 for(;isdigit(ch);ch=getchar())x=x*10+ch-'0';return x*f; 19 } 20 21 const int N = 200005; 22 23 int a[N], q[N], b[N], pos[N]; 24 vector<int> T[N]; 25 bool vis[N]; 26 27 bool cmp(int i,int j) { 28 return pos[i] < pos[j]; 29 } 30 31 int main() { 32 int n = read(); 33 for (int i=1; i<n; ++i) { 34 int u = read(), v = read(); 35 T[u].push_back(v), T[v].push_back(u); 36 } 37 for (int i=1; i<=n; ++i) a[i] = read(), pos[a[i]] = i; 38 for (int i=1; i<=n; ++i) 39 sort(T[i].begin(), T[i].end(), cmp); 40 int L = 1, R = 0, tot = 0; 41 q[++R] = 1; 42 while (L <= R) { 43 int u = q[L ++]; 44 b[++tot] = u; 45 vis[u] = true; 46 for (int sz=T[u].size(),i=0; i<sz; ++i) { 47 int v = T[u][i]; 48 if (vis[v]) continue; 49 q[++R] = v; 50 } 51 } 52 for (int i=1; i<=n; ++i) { 53 if (a[i] != b[i]) { 54 puts("No"); return 0; 55 } 56 } 57 puts("Yes"); 58 return 0; 59 }