并查集(Union-Find)是一种数据结构,它提供了处理一些不交集的合并及查询问题的高效方法。并查集主要支持两种操作:

查找(Find):确定某个元素属于哪个子集,这通常意味着找到该子集的“代表元素”或“根元素”。

合并(Union):将两个子集合并成一个集合。

并查集通过数组或树形结构来实现,其中每个节点指向其父节点,根节点指向自身,这样形成一个或多个树形结构。每棵树代表一个集合,树根的标识符(通常是数组的索引)代表整个集合的标识符。

基本概念:
初始化:开始时,每个元素各自构成一个单元素集合,即每个元素的父节点是其自身。
路径压缩:在执行查找操作时,将查找路径上的每个节点直接连接到根节点,这样可以加快后续查找的速度。
按秩合并:合并时,总是将更小的树连接到更大的树的根节点上,这可以帮助避免树变得过深,从而保持操作的效率。

并查集的重要思想在于,用集合中的一个元素代表集合。
【数据结构-图论】并查集-LMLPHP
现在1号和3号比武,假设1号赢了(这里具体谁赢暂时不重要),那么3号就认1号作帮主(合并1号和3号所在的集合,1号为代表元素)。
【数据结构-图论】并查集-LMLPHP
现在2号想和3号比武(合并3号和2号所在的集合),但3号表示,别跟我打,让我帮主来收拾你(合并代表元素)。不妨设这次又是1号赢了,那么2号也认1号做帮主。
【数据结构-图论】并查集-LMLPHP
上面大概介绍完了整体的东西下面介绍一下细节:
【数据结构-图论】并查集-LMLPHP
下面是代码部分:

// 查找i的代表元素,并进行路径压缩优化
int find(int i) {
    if (fa[i] == i)  // 如果元素i指向自己,那么它是代表元素
        return i;
    else
        return fa[i] = find(fa[i]);  // 否则递归查找,并更新i的父链接为代表元素
}

// 合并i和j所在的集合
void unionn(int i, int j) {
    int i_fa = find(i);  // 查找i的代表元素
    int j_fa = find(j);  // 查找j的代表元素
    fa[i_fa] = j_fa;     // 将i的集合合并到j的集合中
}

find 函数通过递归查找找到一个元素的代表元素,并在查找的过程中将元素直接链接到代表元素,这个优化叫做路径压缩,它可以减少后续查找的时间。

unionn 函数将两个元素所在的集合合并成一个集合。它首先找到每个元素的代表元素,然后将其中一个集合的代表元素链接到另一个集合的代表元素上,从而完成合并。这里没有实现按秩合并或路径压缩的更复杂的优化。

下面是一道题
【数据结构-图论】并查集-LMLPHP

public class UnionFind {
    private int[] parent;

    public UnionFind(int n) {
        parent = new int[n];
        for (int i = 0; i < n; i++) {
            parent[i] = i;
        }
    }

    public int find(int x) {
        if (x != parent[x]) {
            parent[x] = find(parent[x]);
        }
        return parent[x];
    }

    public void union(int x, int y) {
        parent[find(x)] = find(y);
    }

    public boolean isConnected(int x, int y) {
        return find(x) == find(y);
    }
    
    public static void main(String[] args) {
     
        UnionFind uf = new UnionFind(10);
        uf.union(0, 1); // Marry person 1 and 2
        uf.union(2, 3); // Marry person 3 and 4
        boolean areMarried = uf.isConnected(1, 4); // Check if person 2 and 5 are related
        System.out.println(areMarried ? "YES" : "NO"); // Output should be "NO" if unrelated
    }
}
03-02 07:26