天天写算法之It is a tree?

题目地址 点击打开链接

这个题遇上一个题目，几乎是一样的，主要的区别就是：

这个题是有向图，而上一个题目为无相图，一开始直接改了一下输出，后来发现有方向，于是只加了一句就过了。也就是

当你将要合并点时，需要判断儿子结点，他是不是已经有祖先了？也就是他的祖先是不是自己。如果不是

说明已经有祖先了，再有想当他祖先的就直接gg

代码如下：

#include <iostream>
#include <string>
#include<string.h>
#include<cstdio>
using namespace std;
#define MAX 111111

int pre[MAX];//父节点数组
int Vis[MAX];
int rk[MAX];//标记所有性别，这里按照等级理解先，等级%2为奇数为一种，偶数为另一种
int n ,n_test,n_relation,cnt;
bool flag ;
void init()
{
    cnt=0;
    for(int i = 1 ; i<MAX;i++)
    {
        pre[i] = i;
        Vis[i]=0;
    }
   flag = false ;
}
int Find(int x ){
    if(pre[x]!=x)
    {
        pre[x] = Find(pre[x]);
    }
    return pre[x];

}

void Union(int a,int b)
{
  int fa=Find(a);
  int fb=Find(b);
   if(fa!=fb)
   {
       pre[fb]=fa;
   }


   if(fa==fb||fb!=b)
   {
      // printf("true");
       flag = true ;
   }
}


int main(){
    int a ,b,index = 1;
    init();
    while(cin>>a >>b)
    {
        if(a==-1&&b==-1)
        {
            return 0;
        }
        if(a==0&&b==0)
        {
            for(int i =1 ; i<=MAX; i++)
            {
                if(Vis[i]&&pre[i]==i)
                {
                    cnt++;
                }
            }
            if(cnt>1)
            {
                flag = true ;
            }
            if(flag)
            {
                printf("Case %d is not a tree.\n",index++);
            }else {
                printf("Case %d is a tree.\n",index++);
            }
            init();
           continue ;
        }
        Vis[a]=Vis[b]=1;
        if(flag==true )
        {
            continue;
        }
        Union(a,b);
    }

}