poj 1308 Is It A Tree?(并查集）

题目： poj 1308 Is It A Tree?(并查集）

题目大意：给出每两个节点的关系，判断这样关系下形成的是树吗？

解题思路：这题要注意：1.空树也是树，2.然后树是一个根节点多个子节点，不能联通，3.也不能边是双向的，4.不能是森林。

然后用并查集就可以排除掉2，3两种不可能情况，但是切忌形成森林，所以这里还需要用set来实现。

代码：

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

const int N = 1e6+5;
int f[N];
set<int> vis;

int getfather (int n) {

	return f[n] = (n == f[n]) ? n: getfather(f[n]);
}

void init () {

	for (int i = 0; i < N; i++)
		f[i] = i;
}

int main () {
	
	int x, y, flag, cas = 0, ori;
	while (scanf ("%d%d", &x, &y)) {
		
	  if (x == -1 && y == -1)
		  break;
		
	  init ();
	  flag = 1;
	  ori = x;
	  while (x || y) {

		int p = getfather(x);
		int q = getfather(y);
		vis.insert(x);
		vis.insert(y);
		if (p != q)
			f[q] = p;
		else 
			flag = 0;
		scanf ("%d%d", &x, &y);
	  }

	  ori = getfather(ori);
	  for (set<int>::iterator i = vis.begin(); i != vis.end(); i++)
		 if (getfather(*i) != ori) {

	    	 flag = 0;
	  	   	break;
		 }
	  if (flag)
		  printf ("Case %d is a tree.\n", ++cas);
	  else
		  printf ("Case %d is not a tree.\n", ++cas);
	  vis.clear();
	}
	return 0;
}