省份数量【并查集】

有 n 个城市，其中一些彼此相连，另一些没有相连。如果城市 a 与城市 b 直接相连，且城市 b 与城市 c 直接相连，那么城市 a 与城市 c 间接相连。

省份 是一组直接或间接相连的城市，组内不含其他没有相连的城市。

给你一个 n x n 的矩阵 isConnected ，其中 isConnected[i][j] = 1 表示第 i 个城市和第 j 个城市直接相连，而 isConnected[i][j] = 0 表示二者不直接相连。

返回矩阵中 省份 的数量。

class Solution {
    void init(int[] pre){
        for(int i=0;i<pre.length;i++){
            pre[i]=i;
        }
    }
    int find(int x,int[] pre){
        while(pre[x]!=x){
            x=pre[x];
        }
        return x;
    }
    //合并
    void merge(int x,int y,int[] pre){
        int a=find(x,pre);
        int b=find(y,pre);
        pre[a]=b;
    }
    public int findCircleNum(int[][] isConnected) {
        int[] pre=new int[isConnected.length];
        init(pre);
        for(int i=0;i<isConnected.length;i++){
            for(int j=0;j<isConnected[i].length;j++){
                if(i!=j){
                    if(isConnected[i][j]==1){
                        merge(i,j,pre);
                   

                    }
                }
            }
        }
        int count=0;
        for(int i=0;i<pre.length;i++){
            if(pre[i]==i){
                count++;
            }
        }
        return count;
    }
}