[Codeforces 1037D] Valid BFS？

[题目链接]

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

[算法]

首先求出每个点的父节点 ， 每棵子树的大小

然后判断BFS序是否合法即可

时间复杂度 ： O(N)

[代码]

#include<bits/stdc++.h>
using namespace std;
const int MAXN = 2e5 + ;

struct edge
{
        int to , nxt;
} e[MAXN << ];

int n , tot;
int a[MAXN],fa[MAXN],head[MAXN],size[MAXN];

template <typename T> inline void chkmax(T &x,T y) { x = max(x,y); }
template <typename T> inline void chkmin(T &x,T y) { x = min(x,y); }
template <typename T> inline void read(T &x)
{
    T f = ; x = ;
    char c = getchar();
    for (; !isdigit(c); c = getchar()) if (c == '-') f = -f;
    for (; isdigit(c); c = getchar()) x = (x << ) + (x << ) + c - '';
    x *= f;
}
inline void addedge(int u,int v)
{
        tot++;
        e[tot] = (edge){v,head[u]};
        head[u] = tot;
}

int main()
{

        read(n);
        if (n == ) { puts("YES"); return ; }
        for (int i = ; i < n; i++)
        {
                int u , v;
                read(u); read(v);
                addedge(u,v);
                addedge(v,u);
        }
        for (int i = ; i <= n; i++) read(a[i]);
        if (a[] != ) { puts("NO"); return ; }
        queue< int > q;
        fa[] = -;
        q.push();
        while (!q.empty())
        {
                int u = q.front();
                q.pop();
                for (int i = head[u]; i; i = e[i].nxt)
                {
                        int v = e[i].to;
                        if (v != fa[u])
                        {
                                size[u]++;
                                fa[v]    = u;
                                q.push(v);
                        }
                }
        }
        int t =  , cnt = ;
        for (int i = ; i <= n; i++)
        {
                if (fa[a[i]] == a[t])
                {
                        cnt++;
                        continue;
                }
                if (cnt == size[a[t]])
                {
                        while (t < i && size[a[t + ]] == ) t++;
                        if (t == i)
                        {
                                printf("NO\n");
                                return ;
                        }
                        cnt = ;
                        t++;
                        continue;
                }
                printf("NO\n");
                return ;
        }
        if (cnt == size[a[t]]) printf("YES\n");
        else printf("NO\n");

        return ;

}