练习题 - 力扣 547. 省份数量

练习题 - 力扣 547. 省份数量
有 n 个城市，其中一些彼此相连，另一些没有相连。如果城市 a 与城市 b 直接相连，且城市 b 与城市 c 直接相连，那么城市 a 与城市 c 间接相连。

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

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

返回矩阵中 省份 的数量。

示例 1：

输入：isConnected = [[1,1,0],[1,1,0],[0,0,1]]
输出：2
示例 2：

输入：isConnected = [[1,0,0],[0,1,0],[0,0,1]]
输出：3

作者：爱学习的饲养员
链接：https://leetcode-cn.com/leetbook/read/graph/rqhz0m/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

class Solution {
public:
    void UnionFind(int root[],int size){
        for(int i=0;i<size;i++){
            root[i]=i;
        }
    }
    int find(int root[],int x){
        if(x==root[x]){
            return x;
        }
        return root[x]=find(root,root[x]);
    }
    
    void Union(int root[],int x,int y){
        int rootX=find(root,x);
        int rootY=find(root,y);
        if(rootX!=rootY){
            root[rootY]=rootX;
        }
    }

    bool connected(int root[],int x, int y) {
        return find(root,x) == find(root,y);
    }

    int findCircleNum(vector<vector<int>>& isConnected) {
        
        int sum=isConnected.size();
        int *root=new int[sum];
        UnionFind(root,sum);

        for(int i=0;i<isConnected.size();i++){
            for(int j=0;j<isConnected.size();j++){
                if(isConnected[i][j]){
                    Union(root,i,j);
                }
            }
        }
        int ans=sum;
        for(int i=0;i<sum;i++){
            if(root[i]!=i){
                ans--;
            }
        }
        return ans;
    }
    
};