[路飞]_547. 省份数量

182 阅读1分钟

题目介绍

有 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  

提示:

1 <= n <= 200 n == isConnected.length n == isConnected[i].length isConnected[i][j] 为 1 或 0 isConnected[i][i] == 1 isConnected[i][j] == isConnected[j][i] 通过次数187,963提交次数303,292

来源:力扣(LeetCode) 链接:leetcode-cn.com/problems/nu… 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路

用并查集的思路解题,其实就是求并查集中有多少个集合

/**
 * @param {number[][]} isConnected
 * @return {number}
 */

class UnionSet {
  constructor(n) {
    this.fa = new Array(n + 1)
    for (let i = 0; i <= n; i++) {
      this.fa[i] = i
    }
  }
  get(x) {
    let fa = this.fa
    // 路径优化
    return fa[x] = (fa[x] === x ? x : (this.get(fa[x])))
  }
  merge(a, b) {
    this.fa[this.get(a)] = this.get(b)
  }
}
var findCircleNum = function(isConnected) {
    let n = isConnected.length
    let unionSet = new UnionSet(n)
    for (let i = 0; i < n; i++) {
        for (let j = 0; j < isConnected[i].length; j++) {
            if (isConnected[i][j] === 1) {
                unionSet.merge(i, j)
            }
        }
    }
    let ans = 0
    for (let i = 0; i < n; i++) {
        if (i === unionSet.get(i)) ans+=1
    }
    return ans
};