POJ 1308 Is It A Tree? 基础并查集★

题目大意就是判断是否为一棵树

注意：0 0空树也是树

然后自己指向自己也是不可以的

 

#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
#include<string>
#include<cstring>
#include<cstdio>
const int maxn=205;
const int INF=0x3f3f3f3f;
typedef long long LL;
using namespace std;
int par[maxn];
int vis[maxn];
void init()
{
	for(int i=0;i<maxn;i++) par[i]=i;
}
int GetRoot(int a)
{
	return par[a]==a?a:par[a]=GetRoot(par[a]);
}

void Merge(int a,int b)
{
	int p1=GetRoot(a);
	int p2=GetRoot(b);
	if(p1!=p2) par[p2]=p1;
}

bool same(int a,int b)
{
	return (GetRoot(a)==GetRoot(b))?true:false; 
}

int main()
{
//	freopen("E:\\ACM\\test.txt","r",stdin);
	int a,b;
	int t=1;
	while(scanf("%d%d",&a,&b),a!=-1&&b!=-1)
	{
		if(a==0&&b==0) {printf("Case %d is a tree.\n",t++);continue;} //空树 
		init();
		memset(vis,0,sizeof(vis));
		int num=0;
		bool flag=true;
		if(a==b) flag=false; //自己指向自己也是不可以的
		else Merge(a,b);
		vis[a]=vis[b]=1;
		int k=a; //k是树中任意一个节点，用来比较这个节点和其他节点是否都在一个集合 
		
		while(scanf("%d%d",&a,&b),a&&b) 
		{
			if(same(a,b)) flag=false; //如果a，b已经在一个集合 
			Merge(a,b);
			vis[a]=vis[b]=1;
		}
		
		for(int i=0;i<maxn;i++) //判断所有的节点是否都在一个集合 
		{
			if(vis[i]&&!same(i,k)) {flag=false;break;}
		}

		if(flag) printf("Case %d is a tree.\n",t++);
        else printf("Case %d is not a tree.\n",t++);
        
	}
	
	return 0;
}