并查集主要用于解决一些元素分组的问题。它管理一系列不相交的集合,并支持两种操作:
- 合并(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 . 初始化
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;
}