LeetCode——547. 省份数量

题目

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

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

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

返回矩阵中 省份 的数量。

思路

简单并查集的应用，其实这个矩阵一定是个对称矩阵，因为两者之间相连了，对应的坐标（i,j）和（j,i）的数值一定是相等的。可以不用全部进行遍历。我们观察了一下，发现说，其实处于主对角线上的数值一定是1，因为自己肯定和自己相连的，也可以不用进行合并。不过为了处理的方便，可以再加一个判断就好了。然后就是依旧并查集两件套，查找父结点，合并集合。

代码

//省份数量
class Solution {
    static int father[]=new int[205]; //一共有200个城市
    public int findCircleNum(int[][] M) {
        int n=M.length;

        //每个城市都是一个省份
        for (int i=0;i<n;++i)
            father[i]=i;

        for (int i=0;i<n;++i)
        {
            for (int j=0;j<n;++j)
            {
                if (M[i][j]==1) //两个城市是互相连接的
                {
                    if (findfather(i)!=findfather(j))
                    {
                        union(i,j);
                    }
                }
            }
        }
        int ans=0;
        for (int i=0;i<n;++i)
            if (father[i]==i)
                ++ans;
        return ans;
    }


    //找到对应的父节点
    private int findfather(int i)
    {
        if (i==father[i])
            return i;
        father[i]=findfather(father[i]); //递归寻找最终父节点，并且进行路径压缩
        return father[i];
    }

    //合并两个城市
    private void union(int i, int j)
    {
        i=findfather(i);
        j=findfather(j);
        if (i!=j)
            father[j]=i;
    }
}

结果

继续熟悉了并查集两件套。