并查集
问题背景
实现强盗之间的归顺算法是一个递归函数,依照“靠左原则”,具体的运行效果程序后面的问题解决过程,程序实现:
int _find(int r) {
if (f[r] == r) return r;
else {
f[r] = _find(r); //路径压缩
return f[r];
}
}
void _union(int i, int j) {
int i_fa = _find(i);
int j_fa = _find(j);
if (i_fa != j_fa) f[j_fa] = i_fa;
return;
}
问题解决过程
第一步:我们申请一个一维数组f,我们用数组下标1~10来表示这10个强盗,用每个下标所对应的单元格来存储每个强盗的BOSS是谁。
第二步:初始化。根据我们之前的约定,这 10 个强盗最开始是各自为政的,每个强盗的BOSS就是自己。"1号强盗"的BOSS就是"1号强盗"自己,因此f【1】的值为1。以此类推,"10号强盗"的BOSS是"10号强盗",即fT101的值为10。请注意这是很重要的一步。
第三步:开始"合并同伙",即如果发现目前两个强盗是同伙,则这两个强盗是同一个犯罪团伙。
例如警方得到的第1条线索是"1号强盗与2号强盗是同伙"。"1号强盗"和"2号强盗"原来的BOSS都是自己,如今发现“1号强盗”和“2号强盗”其实是同一个犯罪团伙,那么究竟是让"1号强盗"变成"2号强盗"的BOSS,还是让"2号强盗"变成"1号强盗"的BOSS呢?一个犯罪团伙只能有一个首领。其实无所谓,都可以。我们这里假定左边的强盗更厉害一些,给这个规定起个名字叫做"靠左"法则。也就是说"2号强盗"的BOSS将变成“1号强盗”。因此我们将f【2】中的数改为1,表明“2号强盗”归顺了“1号强盗”。
线索2:“3号强盗与4号强盗是同伙”
线索3: “5号强盗与2号强盗是同伙”
线索4: “4号强盗与6号强盗是同伙”
线索5: “2号强盗与6号强盗是同伙”
线索6: “8号强盗与7号强盗是同伙”
线索7: “9号强盗与7号强盗是同伙”
线索8: “1号强盗与6号强盗是同伙”
线索9: “2号强盗与4号强盗是同伙”
对于线索9,上图不会发生变化,属于多余的线索
最终的代码实现
#include <bits/stdc++.h>
using namespace std;
int f[101] = {0};
int _find(int r) {
if (f[r] == r) return r;
else {
f[r] = _find(f[r]); //路径压缩
return f[r];
}
}
void _union(int i, int j) {
int i_fa = _find(i);
int j_fa = _find(j);
if (i_fa != j_fa) f[j_fa] = i_fa;
return;
}
int main() {
//初始化
int n=10;
for (int i = 1; i <= n; i++) {
f[i] = i; //i的boss就是i本身
}
//9条线索,合并团伙
_union(1,2);
_union(3,4);
_union(5,2);
_union(4,6);
_union(2,6);
_union(8,7);
_union(9,7);
_union(1,6);
_union(2,4);
//统计团伙个数
int sum=0;
for(int i=1;i<=n;i++){
if(f[i]==i) sum++;
}
//输出结果
cout<<sum;
return 0;
}
总结:
(1) 并查集可以是一个森林(由多课树组成)
(2) 并查集主要是处理一些不相交集合问题
(3) 并查集的使用场景如求连通子图,求最小生成树kruskal算法,求最近公共祖先(LCA)
(4) 并查集的使用过程主要包含三步:初始化init 查询find 合并union