并查集模板

简介

并査集的英文为 Disjoint Set，直译为“不相交集合”。并查集包含了 3项内容:并、查、集。并查集是不相交集合上的合并、查询。
并查集是一种非常精巧而实用的数据结构，在算法竞赛中很常见，原因有3个:一是它简单且高效，二是其应用很直观，三是它容易与其他数据结构和算法结合。并查集的经典应用包括:检查连通性、最小生成树 Kruskal 算法、LCA 算法等。

基本操作

初始化

每个节点的父节点指向自己

void init(int n)
{
    for (int i = 1; i <= n; ++i)
    {
        fa[i] = i;
        rank[i] = 1;
    }
}

查询

一层层访问父节点，直至查到根节点

int find(int x)
{
    if(fa[x] == x)
        return x;
    else
        return find(fa[x]);
}

合并

void merge(int i, int j)
{
    int x = find(i), y = find(j);    //先找到两个根节点
    if (rank[x] <= rank[y])
        fa[x] = y;
    else
        fa[y] = x;
    if (rank[x] == rank[y] && x != y)
        rank[y]++;                   //如果深度相同且根节点不同，则新的根节点的深度+1
}

LeetCode实例（547. 省份数量）

有 n 个城市，其中一些彼此相连，另一些没有相连。如果城市 a 与城市 b 直接相连，且城市 b 与城市 c 直接相连，那么城市 a 与城市 c 间接相连。
省份 是一组直接或间接相连的城市，组内不含其他没有相连的城市。
给你一个 n x n 的矩阵 isConnected ，其中 isConnected[i][j] = 1 表示第 i 个城市和第 j 个城市直接相连，而 isConnected[i][j] = 0 表示二者不直接相连。
返回矩阵中 省份 的数量。

class Solution {
    public int findCircleNum(int[][] isConnected) {
        int n=isConnected[0].length;
        int []fa=new int[n];
        int []rank=new int[n];
        for(int i=0;i<n;i++){
            fa[i]=i;
            rank[i]=1;
        }
        for(int i=0;i<n;i++){
            for(int j=i+1;j<n;j++){
                if(isConnected[i][j]==1)
                    merge(i,j,fa,rank);
            }
        }
        HashSet<Integer>h=new HashSet<>();
        for(int i=0;i<n;i++){
            h.add(find(i,fa));
        }
        return h.size();
    }
    public int find(int x,int []fa){
        if(x==fa[x])
            return x;
        else
            return find(fa[x],fa);
    }
    public void merge(int i,int j,int []fa,int []rank){
        int x=find(i,fa);
        int y=find(j,fa);
        if(rank[x]<=rank[y]){
            fa[x]=fa[y];
        }else
            fa[y]=fa[x];
        if(rank[x]==rank[y]&&x!=y){
            rank[y]++;
        }
    }
}