题解 CF1037D 【Valid BFS?】

不管怎么说，这都不是道紫题吧。。。

---

 

这里采用的思想有点类似轻重链剖分。

我们按照每个节点在序列里面出现的顺序，把每一个节点连出去的边都排一个序。

这样（如果序列没错）肯定会按照序列的方式遍历完全图。

所以我们就按照这个图跑一遍BFS。

一边BFS一边查看是否符合。

程序的速度不算特别快，最优解rank11。（写丑了）

---

 

AC代码如下：

1925ms 9776kb

#include<bits/stdc++.h>

using namespace std;

namespace StandardIO{

    template<typename T>inline void read(T &x){
        x=0;T f=1;char c=getchar();
        for(;c<'0'||c>'9';c=getchar())if(c=='-')f=-1;
        for(;c>='0'&&c<='9';c=getchar())x=x*10+c-'0';
        x*=f;
    }

    template<typename T>inline void write(T x){
        if(x<0)putchar('-'),x*=-1;
        if(x>=10)write(x/10);
        putchar(x%10+'0');
    }

}

using namespace StandardIO;

namespace Solve{
    
    const int N=200200;
    int n,cnt;
    int a[N],b[N],vis[N];
    vector<int>G[N];
    
    inline bool cmp(int x,int y){
        return a[x]<a[y];
    }
    
    inline void solve(){
        read(n);
        for(register int i=1;i<=n-1;++i){
            int u,v;
            read(u),read(v);
            G[u].push_back(v),G[v].push_back(u);
        }
        for(register int i=1;i<=n;++i){
            read(b[i]),a[b[i]]=i;
        }
        for(register int i=1;i<=n;++i){
            sort(G[i].begin(),G[i].end(),cmp);
        }
        queue<int>q;
        q.push(1);
        while(!q.empty()){
            int now=q.front();q.pop();
            if(vis[now])continue;
            vis[now]=1;
            if(b[++cnt]!=now){
                puts("No");
                return;
            }
            for(register int i=0;i<G[now].size();++i){
//                cout<<G[now][i]<<endl;
                q.push(G[now][i]);
            }
        }
        puts("Yes");
    }
}

using namespace Solve;

int main(){
//    freopen(".in","r",stdin);
//    freopen(".out","w",stdout);
    solve();
}