HOJ 1325 Is It A Tree (并查集)

注意
不能成环；
节点的出度只能为0或1：0是根节点，1是子节点；
只能有一个根节点。

#include<iostream>
#include<cstring>
using namespace std;
const int NN = 1000 + 5;

typedef struct node
{
	int a;
	struct node *p;//指向父节点
}node;

node tree[NN];
int vis[NN];//0表示没有数据，-1表示根节点，1表示子节点

int find(int x)
{//返回x根节点的值
	if (tree[x].p == NULL) return x;//若x是根节点则返回自己的值
	return find((*(tree[x].p)).a);
}

int link(int x, int y)
{//把tree[y]变成tree[x]的子节点
	if (vis[y] == 1 || find(x) == y) return 0;//表明y已经有父节点或者x的父节点是y，前者不符合树的定义，后者会形成环
	if (!vis[x]) vis[x] = -1;//-1表明有x数据但它是根节点
 	vis[y] = 1;//标记为1表明y拥有父节点
	tree[x].a = x;
	tree[y].a = y;
	tree[y].p = &tree[x];
	return 1;
}

int main()
{
	int x, y, flag, count = 1;
	while (cin >> x >> y)
	{
		if (x < 0 && y < 0) break;
		if (!x && !y)
		{
			cout << "Case " << count++ << " is a tree." << endl;
			continue;
		}
		flag = 1;
		memset(&tree, 0, sizeof(tree));
		memset(vis, 0, sizeof(vis));
		link(x, y);//y的父节点指针指向x
		while (cin >> x >> y)
		{
			if (!x && !y) break;
			if (link(x, y) == 0)//表明y已经有父节点了或者是x的父节点是y
			{
				flag = 0;
				while (cin >> x >> y) if (!x && !y) break;
				break;                           
			}
		}
		int bb = 0;
		if (flag)//flag初始化为1，若此处已经是0表示上面已经判断完不能满足条件，不再进行判断
			for (int i = 0; i < NN; i++)
			{//via[i]值为-1表示根节点，判断有多少个根节点，一棵树只能有一个，否则判断完毕，不止一棵树
				if (vis[i] < 0) bb++;
				if (bb > 1) { flag = 0; break; }
			}
		if(flag)
			cout << "Case " << count++ << " is a tree." << endl;
		else 
			cout << "Case " << count++ << " is not a tree." << endl;
	}
	return 0;
}