题目介绍
有 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
};