本文实例为大家分享了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