HDU-5423 Rikka with Tree。树深搜

Rikka with Tree

 题意：给出树的定义，给出树相似的定义和不同的定义，然后给出一棵树，求是否存在一颗树即和其相似又与其不同。存在输出NO，不存在输出YES。

 思路：以1号节点为根节点，我们观察到一颗树如果不存在这种树即与其相似又与其不同，那么这棵树要么所有节点的深度都不一样，要么有深度一样的并且他们的父节点都一样，后来想想，如果当前节点的父节点有多个儿子，那么当前节点则不能有儿子节点，否则就可以将儿子节点转移到兄弟节点上。思路很正确，比赛的时候用深搜递归写就错了，改成两层循环枚举任意两个节点看是否符合情况然后就过了，，过了。。代码能力还是不行啊，思路都一样，哪里写错了都不知道。

   来看两种写法：

    深搜标记：WA？

void init(){    memset(d,0,sizeof(d));    memset(vis1,0,sizeof(vis1));     memset(vis2,0,sizeof(vis2));     for(int i=0;i<N;i++) g[i].clear();}void dfs(int u,int fa){    d[u]=d[fa]+1;    if(vis1[d[u]])    {        if(vis1[d[u]]!=fa)//度相同根节点不同那么可以互换        {f=1;//         printf("%d %d %d %d\n",u,fa,d[u],vis1[d[u]]);        }    }    else vis1[d[u]]=fa;    int len=g[u].size();    if(len>1) vis2[u]=1;    for(int i=0;i<len;i++)    {        int v=g[u][i];        if(v!=fa)        {//           printf("%d %d %d %d %d\n",v,u,fa,d[u],vis1[d[u]]);             if(u!=fa&&vis2[fa]) f=1;             dfs(v,u);        }    }}int main(){    int n,u,v;    while(~scanf("%d",&n))    {        init();        for(int i=1;i<n;i++)        {            scanf("%d%d",&u,&v);            g[u].push_back(v);            g[v].push_back(u);        }        f=0;        dfs(1,1);        if(f) puts("NO");        else puts("YES");    }    return 0;}

  以上有种样例能debug：

5

1 2

2 3

3 4

3 5

output: NO

Answer : YES

  深搜预处理+枚举，AC！正确思路：父节点一样的情况下不能有儿子节点，否则可以将儿子节点转移至兄弟节点。

void init(){    memset(pre,0,sizeof(pre));    memset(d,0,sizeof(d));    memset(vis1,0,sizeof(vis1));    memset(vis2,0,sizeof(vis2));    for(int i=0; i<N; i++) g[i].clear();}void dfs(int u,int fa){    d[u]=d[fa]+1;    pre[u]=fa;    int len=g[u].size();    for(int i=0; i<len; i++)    {        int v=g[u][i];        if(v!=fa) dfs(v,u);    }}int main(){    int n,u,v;    while(~scanf("%d",&n))    {        init();        for(int i=1; i<n; i++)        {            scanf("%d%d",&u,&v);            g[u].push_back(v);            g[v].push_back(u);        }        f=0;        dfs(1,1);        for(int i=1; i<=n&&!f; i++)            for(int j=1; j<=n&&!f; j++)                if(i!=j&&d[i]==d[j])                {                    if(pre[i]==pre[j])                    {                        int k1=g[i].size();                        int k2=g[i].size();                        if(k1>1||k2>1) f=1;                    }                    else f=1;                }        if(f) puts("NO");        else puts("YES");    }    return 0;}