畅通工程(并查集)

畅通工程

某省调查城镇交通状况，得到现有城镇道路统计表，表中列出了每条道路直接连通的城镇。省政府“畅通工程”的目标是使全省任何两个城镇间都可以实现交通（但不一定有直接的道路相连，只要互相间接通过道路可达即可）。问最少还需要建设多少条道路？
Input
测试输入包含若干测试用例。每个测试用例的第1行给出两个正整数，分别是城镇数目N ( < 1000 )和道路数目M；随后的M行对应M条道路，每行给出一对正整数，分别是该条道路直接连通的两个城镇的编号。为简单起见，城镇从1到N编号。
注意:两个城市之间可以有多条道路相通,也就是说
3 3
1 2
1 2
2 1
这种输入也是合法的
当N为0时，输入结束，该用例不被处理。
Output
对每个测试用例，在1行里输出最少还需要建设的道路数目。
Sample Input
4 2
1 3
4 3
3 3
1 2
1 3
2 3
5 2
1 2
3 5
999 0
0
Sample Output
1
0
2
998

思路分析：

这道题是并查集的经典例题。
并查集算法的操作是：
1.查找两个元素是否在同一集合。
2.将两个元素所在的集合合并起来。
因此并查集常用于处理不相交集合的合并以及查询任意两元素是否为同一集合。在这一道题中，两个城镇可以连通可以看作为这两个结点在一个集合里，问还需要建设多少条道路就是集合数-1.

代码如下：

#include <iostream>
#define MAX 1010
using namespace std;

int pre[MAX];	//pre下标为元素值，pre值为父结点下标 
int h[MAX];		//集合树的高度 
int N, M; 

void init()		//初始化集合，每个元素的父结点为该元素本身，并且高度为1 
{
	for(int i = 0;i <= N;i++)
	pre[i] = i, h[i] = 1;
}

int Find(int x)		//查找所在集合的根结点 
{
	if(pre[x] == x) return x;
	return pre[x] = Find(pre[x]);	//空间压缩，使集合内的所有结点直接指向根结点，减少树的高度 
}

void Union(int x, int y)	//并集 
{
	int rx = Find(x);		//找到这两个元素的根结点 
	int ry = Find(y);
	if(pre[rx] == pre[ry]) return;	//根结点相同，表示在同一集合，无需并 
	if(h[rx] > h[ry]) pre[ry] = rx;	//将高度小的集合树并到高度大的集合树 
	else if(h[rx] < h[ry]) pre[rx] = ry;
	else pre[ry] = rx, h[rx]++;		//集合高度相同，随便并一个，并且高度+1 
}

int main()
{
	int x, y, ans;
	while(~scanf("%d",&N) && N)
	{
		scanf("%d",&M);
		init();
		ans = 0;
		while(M--)
		{
			scanf("%d%d",&x,&y);
			Union(x,y);
		}
		for(int i = 1;i <= N;i++)	//统计集合数(统计集合树根结点的个数) 
		{
			if(pre[i] == i) ans++;
		} 
		printf("%d\n",ans-1);
	}
	return 0;
}