题目来源: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
提示:
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]
代码实现
/**
* @param {number[][]} isConnected
* @return {number}
*/
var findCircleNum = function(isConnected) {
const len = isConnected.length
const uf = new UinoFind(len)
for(let i = 0; i < len; i++){
for(let j = i + 1; j < len; j++){
// 只有俩个城市相连,才去合并
if(isConnected[i][j] == 1){
uf.unite(i,j)
}
}
}
// 连通分量就是我们最后求得省份数量
return uf.getCount()
};
class UinoFind {
constructor(n){ // 初始化并查集里面所有节点的父节点都是它自己
this.parent = new Array(n).fill(0).map((value,index) => index)
this.rank = new Array(n).fill(1) // 集合
this.setCount = n // 连接量
}
findSet(index){// 快速查找
if(this.parent[index] != index){
this.parent[index] = this.findSet(this.parent[index])
}
return this.parent[index]
}
unite(index1,index2){// 合并:合并两个顶点坐标
// 获取俩个顶点坐标
let root1 = this.findSet(index1)
let root2 = this.findSet(index2)
if(root1 != root2){
if(root1 < root2){
[root1,root2] = [root2,root1]
}
// 合并根节点
this.parent[root2] = root1
// 计算已经合并的节点的数量
this.rank[root1] +=this.rank[root2]
// 合并一个,城市数量减1
this.setCount--
}
}
getCount(){
return this.setCount
}
connected(index1,index2){ // p判断俩顶点坐标,是否是一个连通分量
let root1 = this.findSet(index1)
let root2 = this.findSet(index2)
return root1 == root2
}
}