NYOJ-129 并查集

这个题基本上是并查集稍微一变， 只是加了一些判断条件而已，就是将点合并成树， 最后遍历一下， 统计一下有多少棵树， 如果不是1的话， 肯定不是树，所以，可以根据这个来判断

 #include <stdio.h>
 #include <string.h>
 #include <algorithm>
 using namespace std;

 const int MAX = ;
 int f[MAX], IN[MAX], k = , sum = ;//IN表示节点的入度
 bool visit[MAX];/*标记数组中的点是否存在， 就是是否是图中的点，
 只要是输入的点， 其visit全部为true*/
 bool flag;//标记是否是树
 //初始化
 void init()
 {
     for(int i = ; i < MAX; i++)
         f[i] = i;
 }
 //找到它的父亲
 int getf(int i)
 {
     if(i != f[i])
     {
         f[i] = getf(f[i]);//路径压缩
     }
     return f[i];
 }
 //合并函数，
 void merge(int i, int j)
 {
     int t1 = getf(i);
     int t2 = getf(j);
     if(t1 != t2)
     {
         f[t2] = t1;
     }
     else
         flag = false;//如果输入的点已经存在
 }

 int main()
 {
     int x, y;
     flag = true;
     int max_num = ;
     memset(f, , sizeof(f));
     memset(IN, , sizeof(IN));
     memset(visit, false, sizeof(visit));
     init();
     while(scanf("%d %d", &x, &y))
     {
         if(x == - && y == -)
             break;
         if(x ==  && y == )
         {
             sum = ;
             for(int i = ; i <= max_num; i++)
             {
                 if(visit[i] && i == f[i])//如果是根节点， 也就是树的个数
                     sum++;
             }
             for(int i = ; i <= max_num; i++)
                 if(visit[i] && IN[i] > )//如果一个节点的入度大于1
                     flag = false;
             if(sum > )
                 flag = false;
             if(flag)
                 printf("Case %d is a tree.\n", ++k);
             else
                 printf("Case %d is not a tree.\n", ++k);
             memset(f, , sizeof(f));
             memset(visit, false, sizeof(visit));
             memset(IN, , sizeof(IN));
             flag = true;
             init();
         }
         else
         {
             if(!flag)
                 continue;
             max_num = max(max_num, max(x, y));
             visit[x] = true;//标记此点在图中
             visit[y] = true;
             IN[y]++;//将第二个点的入度加一
             merge(x, y);
         }
     }
     return ;
 }