hdu5423

链接：点击打开链接

题意：两棵树中所有点到根节点的距离都相同时这两棵树叫做相似的树，两棵树中有某个节点的父节点不同时这两棵树叫不同的数，如果一棵树不存在另一棵树与之不同且相似则称这棵树为特殊的树，给出一棵树判断是否为特殊的树

代码：

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
int s[1005][1005],sum[1005],dis[1005];
int n;
void dfs(int x,int num){
    int i;
    dis[x]=num;
    for(i=1;i<=n;i++)
    if(s[x][i]&&!dis[i])
    dfs(i,num+1);
}                                       //dfs求出每个节点的深度
int main(){
    int i,a,b,ans;
    while(scanf("%d",&n)!=EOF){
        memset(s,0,sizeof(s));
        memset(sum,0,sizeof(sum));
        memset(dis,0,sizeof(dis));
        for(i=1;i<=n-1;i++){
            scanf("%d%d",&a,&b);
            s[a][b]=s[b][a]=1;
        }
        dfs(1,1);
        for(i=1;i<=n;i++)
        sum[dis[i]]++;
        ans=0;
        for(i=1;i<=n;i++){
            if(sum[i]==1)
            ans++;
            else{
              if(sum[i+1]==0)
              ans=n;
              break;
            }                               //只有当每个深度都只有一个节点或者
        }                                   //只有最深的那一层节点不为1时才是
        if(ans==n)                          //特殊的数
        printf("YES\n");
        else
        printf("NO\n");
    }
    return 0;
}