D. Valid BFS?(思维)

最新推荐文章于 2024-07-27 22:24:45 发布

小菜鸡加油

最新推荐文章于 2024-07-27 22:24:45 发布

阅读量291

点赞数

分类专栏：思维 # bfs

本文链接：https://blog.csdn.net/zstuyyyyccccbbbb/article/details/113744518

版权

思维同时被 2 个专栏收录

514 篇文章 4 订阅

订阅专栏

bfs

13 篇文章 1 订阅

订阅专栏

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

题意：给出一颗以1号节点为根的树，以及一个序列，问所给序列是不是所给树的一种BFS搜索顺序；

思路:开始单纯地判断dfs处理出来的depth了。但是bfs当前出队列的点还和其父亲先进后进有关。

如果考虑存下所有可能顺序，再去比较显然很难实现。

转换思路，根据所给出的序列去构造搜索序列，如果矛盾了，说明所给序列不是一种。

如果构造，给定序列已经有了，每一个都标记优先级，将原先建好的边按照优先级进行sort,赋予新的顺序。再进行bfs看此时出现的点是否对应给定序列。

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=2e5+100;
typedef long long LL;
inline LL read(){LL x=0,f=1;char ch=getchar();	while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
vector<LL>g[maxn];
LL id[maxn],a[maxn],n;
bool vis[maxn];
bool cmp(LL A,LL B){
    return id[A]<id[B];
}
void bfs(){
    queue<LL>que;
    LL s=1;que.push(s);vis[s]=true;
    LL tot=0;
    while(!que.empty()){
        LL u=que.front();que.pop();
        if(u!=a[++tot]){
            cout<<"NO"<<"\n";return;
        }
        for(LL i=0;i<g[u].size();i++){
            LL v=g[u][i];
            if(vis[v]) continue;
            vis[v]=true;
            que.push(v);
        }
    }
    cout<<"YES"<<"\n";
    return;
}
int main(void)
{
  cin.tie(0);std::ios::sync_with_stdio(false);
  cin>>n;
  for(LL i=1;i<n;i++){
    LL u,v;cin>>u>>v;
    g[u].push_back(v);g[v].push_back(u);
  }
  for(LL i=1;i<=n;i++) cin>>a[i];
  for(LL i=1;i<=n;i++){
    id[a[i]]=i;
  }
  for(LL i=1;i<=n;i++){
    sort(g[i].begin(),g[i].end(),cmp);
  }
  bfs();
return 0;
}