并查集2

集合之并查集2

1、并查集模板

并查集解决的问题很典型，可以归纳出模板，方便解题。这个模板包含三个并查集的主要操作：
void Build(int n) ——建立并查集
int Root(int x)——查询根节点
void Unite(int x, int y)——合并并查集
其中Root函数利用递归向上查找根结点，回溯时改变当前结点的父结点，指向根结点；Unite函数查询两个结点的根结点，若根结点相同无需合并，若不同则将合并后的根节点设置为两个结点中秩较大结点的根结点，若两个结点的秩大小也相同，则把后一个结点的根结点设置为合并后的根结点。代码如下：

struct Node {
 	int rnk, pre;
 	void init(int x) {
 		rnk = 0, pre = x;
 	}	
} T[MAX_N];
 
void Build(int n) {
 	for (int i = 1; i <= n; i++) T[i].init(i);
}
 
int Root(int x) {
 	if (T[x].pre != x) T[x].pre = Root(T[x].pre);
 	return T[x].pre;
}
 
void Unite(int x, int y) {
 	x = Root(x), y = Root(y);
 	if (x == y) return;
 	if (T[x].rnk > T[y].rnk) T[y].pre = x;
 	else {
 		T[x].pre = y;
		if (T[x].rnk == T[y].rnk) T[y].rnk++;
 	}
}
 
bool Same(int x, int y) {
 	return Root(x) == Root(y);
}

2、题目——村村通

题目描述
某市调查城镇交通状况，得到现有城镇道路统计表。表中列出了每条道路直接连通的城镇。市政府 “村村通工程” 的目标是使全市任何两个城镇间都可以实现交通（但不一定有直接的道路相连，只要相互之间可达即可）。请你计算出最少还需要建设多少条道路？
输入格式
输入包含若干组测试测试数据，每组测试数据的第一行给出两个用空格隔开的正整数，分别是城镇数目 nn 和道路数目 mm ；随后的 mm 行对应 mm 条道路，每行给出一对用空格隔开的正整数，分别是该条道路直接相连的两个城镇的编号。简单起见，城镇从 11 到 nn 编号。
注意：两个城市间可以有多条道路相通。
输出格式
对于每组数据，对应一行一个整数。表示最少还需要建设的道路数目。
输入输出样例
输入
4 2
1 3
4 3
3 3
1 2
1 3
2 3
5 2
1 2
3 5
999 0
0
输出
1
0
2
998
解题过程：
这道题很明显用并查集做，两个村子至少需要一条路，三个村子至少需要两条路，那么我们把已通路的村子称为集合，两个集合至少需要一条路，三个集合至少需要两条路，所以这道题考察的就是通路后有多少个并查集ans，并输出ans-1，有了前面的并查集模板，这道题就很简单了。
建立一个数组给根结点做标记，最后遍历一遍数出有多少个根结点即是并查集数目ans，并输出ans-1（其实我这么做麻烦了，直接判断每个结点的根结点是否为本身即可，若为本身即代表是根结点，只用一次循环）：

int count[1005] = {0},ans=0;
for(int i=1;i<=n;i++){
	count[Root(i)] = 1;
}
for(int i=1;i<=n;i++){
	if(count[i]==1) ans++;
}
cout << ans-1 << endl;

全部代码：

#include<bits/stdc++.h>
#define MAX_N 5005
using namespace std;
struct Node {
 	int rnk, pre;
 	void init(int x) {
 		rnk = 0, pre = x;
 	}	
} T[MAX_N];
 
void Build(int n) {
 	for (int i = 1; i <= n; i++) T[i].init(i);
}
 
int Root(int x) {
 	if (T[x].pre != x) T[x].pre = Root(T[x].pre);
 	return T[x].pre;
}
 
void Unite(int x, int y) {
 	x = Root(x), y = Root(y);
 	if (x == y) return;
 	if (T[x].rnk > T[y].rnk) T[y].pre = x;
 	else {
 		T[x].pre = y;
		if (T[x].rnk == T[y].rnk) T[y].rnk++;
 	}
}
 
bool Same(int x, int y) {
 	return Root(x) == Root(y);
}
int main(){
	int n,m,x,y;
	
	while(scanf("%d",&n)!=EOF && n != 0){
		scanf("%d",&m);
		Build(n);
		while(m--){
			cin >> x >> y;
			Unite(x,y);
		}
		int count[1005] = {0},ans=0;
		for(int i=1;i<=n;i++){
			count[Root(i)] = 1;
		}
		for(int i=1;i<=n;i++){
			if(count[i]==1) ans++;
		}
		cout << ans-1 << endl;
	}
	return 0;
}