hdu 5423 Rikka with Tree DFS 或最短路（搜索）

Rikka with Tree Accepts: 207 Submissions: 815

这个好像是转载的，忘了原文地址了，改成原创了，有发现的可以提醒下

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 65536/65536 K (Java/Others)
问题描述
众所周知，萌萌哒六花不擅长数学，所以勇太给了她一些数学问题做练习，其中有一道是这样的：

对于一棵树TT,令F(T,i)F(T,i)为点1到点ii的最短距离（边长是1）.

两棵树AA和BB是相似的当且仅当他们顶点数相同且对于任意的ii都有F(A,i)=F(B,i)F(A,i)=F(B,i).

两棵树AA和BB是不同的当且仅当他们定点数不同或者存在一个ii使得以1号点为根的时候ii在两棵树中的父亲不同。

一棵树AA是特殊的当且仅当不存在一棵和它不同的树和它相似。

现在勇太想知道一棵树到底是不是特殊的。

当然，这个问题对于萌萌哒六花来说实在是太难了，你可以帮帮她吗？
输入描述
数据组数不超过100组。每组数据的第一行一个整数n(2 \leq n \leq 1000)n(2≤n≤1000)。

接下来n-1n−1行。每行两个整数u,v(1 \leq u,v \leq n)u,v(1≤u,v≤n)，代表给定树上的一条边。
输出描述
对于每一组数据，如果给定树是特殊的输出"YES"否则输出"NO"。
输入样例
3
1 2
2 3
4
1 2
2 3
1 4
输出样例
YES
NO
Hint
对于第二组数据下面这棵树和它相似。
4
1 2
1 4
3 4

由题可知，所给的树必须是一个直线型的，但是最后一个父亲可以有分叉。
所以可以dfs求出每个点到1这个根的最短距离，然后判断最短距离是否为一即可，但最长的最短距离可以有多个。
因为出现了1个点到多个点的最短路问题，所以可以用最短路的几种解法求解。

#include<iostream>
#include<vector>
#include<cstdio>
#include<cstring>
using namespace std;
vector<int>q[1024];
int dis[1024],leg[1024];

void dfs(int u,int leg)
{
    dis[u]=leg;
    for(int i=0; i<q[u].size(); i++)
    {
        if(dis[q[u][i]]==-1)
            dfs(q[u][i],leg+1);
    }
}
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        int a,b;
        for(int i=1; i<=n; i++)
            q[i].clear();
        memset(dis,-1,sizeof(dis));
        memset(leg,0,sizeof(leg));
        for(int i=1; i<=n-1; i++)
        {
            scanf("%d%d",&a,&b);
            q[a].push_back(b);
            q[b].push_back(a);
        }
        dfs(1,0);
        for(int i=1; i<=n; i++)
            if(dis[i]!=-1)
                leg[dis[i]]++;
        bool mark=true;
        bool mark1=true;
        for(int i=n; i>=1; i--)
        {
            if(leg[i])
            {
                if(mark) mark=false;
                else if(leg[i]>1)
                {
                        mark1=false;
                        break;
                }

            }
        }
        if(mark1) cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }
    return 0;
}

参考博客：
http://blog.csdn.net/xinag578/article/details/48132629