Manthan, Codefest 18 (rated, Div. 1 + Div. 2) D.Valid BFS?(dfs+思维)

最新推荐文章于 2020-03-18 21:38:01 发布

原创最新推荐文章于 2020-03-18 21:38:01 发布 · 165 阅读

0 ·

CC 4.0 BY-SA版权

DFS 专栏收录该内容

7 篇文章

订阅专栏

本文详细介绍了使用DFS深度优先搜索算法进行图遍历的过程，并通过实际代码实现展示了如何记录节点间的关系以及验证路径的正确性。

题解：刚开始把所有点离点1 的距离求出来，并且纪录每个距离的点数，然后逆序依次判断每一个点，是否符合当前距离，如果枚举完了当前距离的点数，就让距离减一，同距离的点判断依据是父节点的出现顺序，我们再找一下就可以了

#include <bits/stdc++.h>
#define inf 0x3f3f3f3f
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;
const int N=2e5+10;
int ar[N],du[N],vis[N],op[N],pre[N],pos[N];
vector<int>G[N];
int n,mx;
void dfs(int x,int cnt)
{
    vis[x]=1;
    du[x]=cnt;
    op[cnt]++;
    mx=max(mx,cnt);
    for(int i=0;i<G[x].size();i++){
        int y=G[x][i];
        if(!vis[y]){
            pre[y]=x;
            dfs(y,cnt+1);
        }
    }
}
int main()
{
    #ifdef local
        freopen("D://rush.txt","r",stdin);
    #endif
    cin>>n;
    mx=1;
    for(int i=0;i<n-1;i++)
    {
        int a,b;
        cin>>a>>b;
        G[a].push_back(b),G[b].push_back(a);
    }
    for(int i=0;i<n;i++)
        cin>>ar[i],pos[ar[i]]=i;
    dfs(1,1); pre[1]=1;
    memset(vis,0,sizeof vis);
    int cnt=mx,last=-1 ;
    for(int i=n-1;i>=0;i--){
        if(du[ar[i]]!=cnt||vis[ar[i]]){cout<<"NO"<<endl;return 0;}
        op[cnt]--,vis[ar[i]]=1;
        int a=pos[pre[ar[i]]];
        if(last==-1) last=a;
        else if(last<a) {cout<<"NO"<<endl;return 0;}
        else last=a ;
        if(op[cnt]==0) cnt--;
    }
    cout<<"YES"<<endl;
    return 0;
}