Manthan, Codefest 18 (rated, Div. 1 + Div. 2) D.Valid BFS?(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;
}
    原文作者:ffgcc
    原文地址: https://www.cnblogs.com/ffgcc/p/10546326.html
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞