本文实例为大家分享了Java实现并查集的具体代码，供大家参考，具体内容如下
自下而上的树结构
接口
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
/**
* @author Nino
*/
public interface UF {
int size();
/**
* 看两个元素是否相连
* @param p
* @param q
* @return
*/
boolean isConnected(int p, int q);
/**
* 将两个元素合并在一起，变成一个集合中的元素
* @param p
* @param q
*/
void unionElements(int p, int q);
}
|
使用路径压缩的并查集
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
/**
* @author Nino
*/
public class UnionFind5 implements UF {
private int[] parent;
//rank[i]表示以i为根的集合中树的层数
private int[] rank;
public UnionFind5(int size) {
parent = new int[size];
rank = new int[size];
for (int i = 0; i < size; i++) {
parent[i] = i;
rank[i] = 1;
}
}
@Override
public int size() {
return parent.length;
}
/**
* 查找过程，查找元素p所对应的集合编号
* O(h)复杂度，h为树的高度
* 使用路径压缩
* @param p
* @return
*/
private int find(int p) {
if (p < 0 && p >= parent.length) {
throw new IllegalArgumentException("p is illegal");
}
if (p != parent[p]) {
parent[p] = find(parent[p]);
}
return parent[p];
}
/**
* 查询p, q是否同属一个集合
* 时间复杂度O(h)
* @param p
* @param q
* @return
*/
@Override
public boolean isConnected(int p, int q) {
return find(p) == find(q);
}
/**
* 合并元素p, q所属的集合，只需要把Rank低的根节点指向Rank高的根节点就可以
* O(h)复杂度
* @param p
* @param q
*/
@Override
public void unionElements(int p, int q) {
int pRoot = find(p);
int qRoot = find(q);
if (pRoot == qRoot) {
return;
}
//败者食尘
if (rank[pRoot] < rank[qRoot]) {
parent[pRoot] = qRoot;
} else if (rank[qRoot] < rank[pRoot]) {
parent[qRoot] = pRoot;
} else {
parent[qRoot] = pRoot;
rank[pRoot] += 1;
}
}
}
|
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持服务器之家。
原文链接：https://blog.csdn.net/Nino_sama/article/details/101028903