POJ 1308 NYOJ 129
View Code
1 /*
2 问题：
3 给你一些有向边 组成一个有向图
4 问这个图是不是一颗树
5 思路：
6 组成树的条件：
7 1.图按无向图计算 是连通的（并查集判断）
8 2.有且仅有一个顶点的入度为零（根节点）
9 3.除根节点外的其余顶点入度必须都为1
10 */
11 #include<iostream>
12 #include<cstring>
13 using namespace std;
14
15 bool flag[10010];//标记
16 int rd[10010];//记录每个点的入度
17
18 int rank[10010];//判断连通性
19 int father[10010];
20
21 int num;//记录顶点个数
22 bool f;
23
24 void init()
25 {
26 int i;
27 num=0;
28 f=0;
29 for(i=0;i<=10000;++i)
30 {
31 flag[i]=0;
32 father[i]=i;
33 rank[i]=1;
34 rd[i]=0;
35 }
36 }
37
38 int find(int x)
39 {
40 if(x!=father[x])
41 father[x]=find(father[x]);
42 return father[x];
43 }
44
45 void uion(int x,int y)
46 {
47 x=find(x);
48 y=find(y);
49 if(x==y)return ;
50 rank[x]+=rank[y];
51 father[y]=x;
52 }
53
54 int main()
55 {
56 int i,x,y,k=1;
57 while(cin>>x>>y&&(x!=-1||y!=-1))
58 {
59 if(!x&&!y)//注意这种情况
60 {
61 cout<<"Case "<<k<<" is a tree."<<endl;
62 k++;
63 continue;
64 }
65 init();
66 if(!flag[x]){num++;flag[x]=1;}
67 if(!flag[y]){num++;flag[y]=1;}
68 rd[y]++;
69 uion(x,y);
70
71 while(cin>>x>>y&&(x||y))
72 {
73 if(!flag[x]){num++;flag[x]=1;}
74 if(!flag[y]){num++;flag[y]=1;}
75 rd[y]++;// 记录每个点的入度数
76 if(rd[y]>1){f=1;continue;} //若有一个点的入度大于1则不是树
77 uion(x,y);
78 }
79
80 int d0=0;//判断入度为零的个数
81 if(!f)
82 {
83 for(i=0;i<=10000;++i)
84 {
85 if(flag[i]&&!rd[i])
86 {
87 d0++;
88 if(d0>1)// 若入度为0 的顶点个数超过1个则 不是树
89 {
90 f=1;
91 break;
92 }
93 }
94 }//for
95 }
96 if(!d0)f=1;// 有且仅有一个顶点的入度为零
97 if(!f)// a判断连通性
98 {
99 for(i=0;i<=10000;++i)
100 {
101 if(flag[i]&&rank[find(i)]!=num)
102 {
103 f=1;//不连通
104 break;
105 }
106 }
107 }
108 if(f)cout<<"Case "<<k<<" is not a tree."<<endl;
109 else cout<<"Case "<<k<<" is a tree."<<endl;
110 k++;
111 }
112 system("pause");
113 return 0;
114 }
115
116
优秀代码：
View Code
1
2 #include <stdio.h>
3
4 #define MAX 10000
5
6 int pre[MAX];
7 int tag[MAX]={0};
8
9 int find_parent(int x)
10 {
11 return pre[x]==x?x:(find_parent(pre[x]));//无压缩路径
12 }
13
14 int main()
15 {
16 int m,n,flag=0,sum=0,count=1,i;
17 for(i=1;i<=MAX;i++)
18 {
19 pre[i]=i;
20 tag[i]=0;
21 }
22 while(scanf("%d%d",&m,&n))
23 {
24 if(m==-1 && n==-1)
25 break;
26 if(m==0 && n==0)
27 {
28 if(flag==0)
29 {
30 for(i=1;i<MAX;i++)
31 {
32 if(tag[i]==0) continue;
33 if(pre[i]==i) sum++; //若连通则只有根节点的祖先是本身
34 }
35 if(sum>1) flag=1;//说明有孤立的点
36 }
37 if(flag==0)
38 printf("Case %d is a tree.\n",count++);
39 else
40 printf("Case %d is not a tree.\n",count++);
41 sum=0;
42 flag=0;
43 for(i=1;i<MAX;i++)
44 {
45 pre[i]=i;
46 tag[i]=0;
47 }
48 continue;
49 }//if(m==0 && n==0)
50 if(flag!=1)
51 {
52 int x,y;
53 x=find_parent(m);
54 y=find_parent(n);
55 tag[m]=1;
56 tag[n]=1;
57 if(x==y) flag=1;
58 else
59 {
60 if(pre[n]==n) pre[n]=x;//边（m,n)，pre[n]==n;表明这是节点n第一次
61 //出现他的父节点m，则m 的祖先就n得祖先
62 // 把pre[n]=x;相当于 找到n最前的祖先
63 else flag=1;//若pre[n]!=n说明n有祖先且不是其现有父节点
64 //m的祖先（因x!=y)说明其入度大于1
65 }
66 }
67 }
68 //system("pause");
69 return 0;
70 }
71
