547. 省份数量

Posted 2021-09-07 易小顺

算法记录

LeetCode 题目：

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

---

思路

• 算法记录
• 说明
• • 一、题目
  • 二、分析
• 总结

---

说明

一、题目

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

二、分析

• 题意就是在遍历整个矩阵，如果对应的点值唯一则两个城市为一个省的。
• 最开始假设城市之间都是独立的，省的数量就是 n ，而一旦确定了是一个省，就可以将实际省的数量进行减一。
• 最后剩余的值就是省的实际数量，也就是整个集合组成的集合的数量。

class Solution {
     class UnionFind{
        int[] father;
        public UnionFind(int n) {
            father = new int[n];
            for(int i = 0; i < n; i++)
                father[i] = i;
        }
        public int find(int x) {
            if(father[x] != x) father[x] = find(father[x]);
            return father[x];
        }
        public int merege(int a, int b) {
            int x = find(a);
            int y = find(b);
            if(x == y) return 0;
            father[x] = y;
            return 1;
        }
    }
    public int findCircleNum(int[][] isConnected) {
        int n = isConnected.length;
        int count = n;
        UnionFind UnionFind = new UnionFind(n);
        for(int i = 0; i < n; i++) {
            for(int j = 0; j < n; j++) {
                if(isConnected[i][j] == 1)
                    count -= UnionFind.merege(i, j);
            }
        }
        return count;
    }
}

---

总结

熟悉并查集的使用方法。