NYOJ-129 并查集

这个题基本上是并查集稍微一变，只是加了一些判断条件而已，就是将点合并成树，最后遍历一下，统计一下有多少棵树，如果不是1的话，肯定不是树，所以，可以根据这个来判断
 1 #include <stdio.h>
 2 #include <string.h>
 3 #include <algorithm>
 4 using namespace std;
 5 
 6 const int MAX = 10050;
 7 int f[MAX], IN[MAX], k = 0, sum = 0;//IN表示节点的入度
 8 bool visit[MAX];/*标记数组中的点是否存在， 就是是否是图中的点，
 9 只要是输入的点， 其visit全部为true*/
10 bool flag;//标记是否是树
11 //初始化
12 void init()
13 {
14     for(int i = 1; i < MAX; i++)
15         f[i] = i;
16 }
17 //找到它的父亲
18 int getf(int i)
19 {
20     if(i != f[i])
21     {
22         f[i] = getf(f[i]);//路径压缩
23     }
24     return f[i];
25 }
26 //合并函数，
27 void merge(int i, int j)
28 {
29     int t1 = getf(i);
30     int t2 = getf(j);
31     if(t1 != t2)
32     {
33         f[t2] = t1;
34     }
35     else
36         flag = false;//如果输入的点已经存在
37 }
38 
39 int main()
40 {
41     int x, y;
42     flag = true;
43     int max_num = 0;
44     memset(f, 0, sizeof(f));
45     memset(IN, 0, sizeof(IN));
46     memset(visit, false, sizeof(visit));
47     init();
48     while(scanf("%d %d", &x, &y))
49     {
50         if(x == -1 && y == -1)
51             break;
52         if(x == 0 && y == 0)
53         {
54             sum = 0;
55             for(int i = 1; i <= max_num; i++)
56             {
57                 if(visit[i] && i == f[i])//如果是根节点， 也就是树的个数
58                     sum++;
59             }
60             for(int i = 1; i <= max_num; i++)
61                 if(visit[i] && IN[i] > 1)//如果一个节点的入度大于1
62                     flag = false;
63             if(sum > 1)
64                 flag = false;
65             if(flag)
66                 printf("Case %d is a tree.
", ++k);
67             else
68                 printf("Case %d is not a tree.
", ++k);
69             memset(f, 0, sizeof(f));
70             memset(visit, false, sizeof(visit));
71             memset(IN, 0, sizeof(IN));
72             flag = true;
73             init();
74         }
75         else
76         {
77             if(!flag)
78                 continue;
79             max_num = max(max_num, max(x, y));
80             visit[x] = true;//标记此点在图中
81             visit[y] = true;
82             IN[y]++;//将第二个点的入度加一
83             merge(x, y);
84         }
85     }
86     return 0;
87 }