并查集 模板题小小讲解

并查集主要用于解决一些元素分组的问题。它管理一系列不相交的集合，并支持两种操作：

• 合并（Union）：把两个不相交的集合合并为一个集合。
• 查询（Find）：查询两个元素是否在同一个集合中。

开讲前我们先来看一个例题：

宗教信仰

世界上有许多宗教，你感兴趣的是你学校里的同学信仰多少种宗教。你的学校有 n 名学生（0 < n ≤ 50000），你不太可能询问每个人的宗教信仰，因为他们不太愿意透露。但是当你同时找到 2 名学生，他们却愿意告诉你他们是否信仰同一宗教，你可以通过很多这样的询问估算学校里的宗教数目的上限。你可以认为每名学生只会信仰最多一种宗教。

输入格式

输入包括多组数据。每组数据的第一行包括 n 和 m，0 ≤ m ≤ n (n−1) / 2，其后 m 行每行包括两个数字 i 和 j，表示学生 i 和学生 j 信仰同一宗教，学生被标号为 1 至 n。

输入以一行 n = m = 0 作为结束。

输出格式

对于每组数据，先输出它的编号（从 1 开始），接着输出学生信仰的不同宗教的数目上限。

Sample Input

10 9
1 2
1 3
1 4
1 5
1 6
1 7
1 8
1 9
1 10
10 4
2 3
4 5
4 8
5 8
0 0

Sample Output

Case 1: 1
Case 2: 7

如题，解题步骤可为：

1. 把信仰同一宗教的人连接成一个集合
2. 查找有多少个集合，即可得出学校内有多少种宗教。

在连接前，我们需要判断要连接的两个人是否已经在同一集合内。如果在，则不需要再次连结。

而上述的 连接 和 判断两值是否在同一集合内 的操作就用到了这篇文章要讲的并查集。

并查集的重要思想在于，用集合中的一个元素代表集合。

所以，可做以下转换：判断两值是否在同一集合内 --> 判断两值所在集合的代表元素是否相等。

这个巨巨写的并查集图文简单易懂：

算法学习笔记(1) : 并查集 - 知乎

大致看了这个巨巨写的图后，应该对并查集理解的差不多了，上模板。

并查集模板：

 1 . 初始化 

void in() {
	for (int i=1;i<=n;i++) {
		pre[i]=i;
	}
	return;
}

 2 . 查找根节点 / 查找该数所在集合的代表值  find()

int find(int a) {
	if (a==pre[a]) return a;
	else return pre[a]=find(pre[a]);  //压缩路径
}

 3 . 连接 uni()

void uni(int a,int b) {
	int q=find(a);
	int w=find(b);
	if (q!=w) {
		pre[q]=w;
	}
	return;
}

完整ac代码

#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;
const int M=1e6+5;
int pre[M];
int n,m,num=1;
void in() {
    for (int i=1;i<=n;i++) {
      	pre[i]=i;
    }
}
int find(int a) {
	if (a==pre[a]) return a;
	else return pre[a]=find(pre[a]);
}
void uni(int a,int b) {
	int q=find(a),w=find(b);
	if (q!=w) {
		pre[q]=w;
	}
	return;
}
int main() {
std::ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
	while (cin>>n>>m,n!=0&&m!=0) {
		in();
		while (m--) {
			int a,b;
			cin>>a>>b;
			uni(a,b);
		}
		int sum=0;
		for (int i=1;i<=n;i++) {
			if (pre[i]==i) sum++;
		}
		printf("Case %d: %d\n",num++,sum);
	}
	return 0;
}