【代码】树之并查集

发表于 更新于 阅读次数：

前言

树之并查集的Java版API实现

源代码

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
package com;

public class UnionFindSets {

    private int[] elementAndGroup;
    private int groupCount;

    public UnionFindSets(int length) {
        // 初始化分组的数量
        groupCount = length;
        // 初始化elementAndGroup数组
        elementAndGroup = new int[length];
        // 初始化elementAndGroup中的元素和所在的标识符
        // 让elementAndGroup的索引作为elementAndGroup的元素值
        for (int i = 0; i < elementAndGroup.length; i++) {
            elementAndGroup[i] = i;
        }
    }

    // 获取并查集分组的个数
    public int count() {
        return groupCount;
    }

    // 获取指定元素的分组标识符
    public int find(int data) {
        return elementAndGroup[data];
    }

    // 检查在并查集中的两个元素是否在相同的组中
    // 检查两个元素所在的树的根节点是否相同
    public boolean connected(int p, int q) {
        return find(p)==find(q);
    }

    // 合并两个不在同一组的元素所在的组进行合并
    public void union(int p, int q) {
        if (!connected(p, q)) {
            return;
        }
        // 找到p所在分组的标识符
        int groupForP = find(p);
        // 找到q所在分组的标识符
        int groupForQ = find(q);
        // 合并
        // 让p所在组的所有元素的标识符改为q所在组的标识符
        for (int i = 0; i < elementAndGroup.length; i++) {
            if (elementAndGroup[i]==groupForP) {
                elementAndGroup[i]=groupForQ;
            }
        }
        // 分组个数自减
        groupCount--;
    }

}

完成

参考文献

哔哩哔哩——黑马程序员