关于并查集的实现思路呢有两种,先讨论第一种
数组实现并查集
首先来看个案例
1 2 3 4 5 6 7 8 9 10
1 2 3 4 5 6 7 8 9 10
这是两串数字
但是,你们也可以当成索引,我自己连我自己,我的索引就是我自己,我对应的数也是我自己。
我们可以把这10个数的当成十个开发者,一开始我全能,我全栈,一个人接私活,我就是我,时间自由
但是这编号 7 的开发者他感觉一个人玩儿没劲,想有几个同事,想有几个伴儿,所以他去人才市场找了家公司,
这个编号 10 的开发者是个老板,这老板一看这 7 挺好啊,能吃能喝能拉能睡,就把他招入麾下,一个公司就有了俩开发者
这时候这十个人的关系就变化了。
1 2 3 4 5 6 7 8 9 10
1 2 3 4 5 6 10 8 9 10
这 7 他对应的就是10了。
这就产生了连接,10我是老板,7是他的下属员工。
产生连接之后我们也得查他到底产没产生连接,我们就得对比这两个的索引位置的值
7 对应的是 10,10 对应的是 10,7和10就产生了连接关系
1 对应的是 1,3对应的是3 , 这两个直接就没产生连接关系
下面用代码体现
private UnionFind1(int n) {
count = n;
id = new int[n];
for (int i = 0; i < n; i++) {
id[i] = i;
}
}
public int find(int n) {
return id[n];
}
public boolean isConnected(int p, int q) {
return find(p) == find(q);
}
//连接
public void unionElement(int p, int q) {
//找到 p 位置的值 pid
int pid = find(p);
//找到 q 位置的值 qid
int qid = find(q);
//假如已经连接就返回
if (isConnected(p, q)) {
return;
}
//将 所有 pid 的索引位置 改为 qid 的值
for (int i = 0; i < count; i++) {
if (id[i] == pid) {
id[i] = qid;
}
}
}
第二种呢,我们把每个数当成一个节点来看待,在初始化方面大同小异啦,主要是在操作方面
1 2 3 4 5 6 7 8 9 10
| | | | | | | | | |
1 2 3 4 5 6 7 8 9 10
这个是一一对应的关系
还是那个故事
7 被 10 招走,7 的 母节点就要变成 10
我们做查询操作的时候,我们也得上溯到母节点,假如10上面还有母节点呢,10的母节点还有母节点呢
所以我们查询的条件为parent[p]=p截止,请细细体会。
UnionFind2(int count) {
parent = new int[count];
this.count = count;
//初始化
//1 2 3 4 5 6 7
//1 2 3 4 5 6 7
for (int i = 0; i < count; i++) {
parent[i] = i;
}
}
//将p节点和q节点进行连接
public void union(int p, int q) {
//寻找p元素的根
int pRoot = find(p);
//寻找q元素节点的根
int qRoot = find(q);
if (pRoot == qRoot) {
return;
}
//相连接
parent[pRoot] = qRoot;
}
public int find(int p) {
while (parent[p] != p) {
p = parent[p];
}
return p;
}
public boolean isConnected(int p, int q) {
return find(p) == find(q);
}