有 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
解题:
class UnionFind {
constructor(length) {
this.parent = new Array(length).fill(0).map((v, index) => index);
this.rank = new Array(length).fill(1);
this.setCount = length;
}
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),
root2=this.findSet(index2);
if(root1!==root2){
if(this.rank[root1]<this.rank[root2]) {
[root1,root2]= [root2,root1]
}
this.parent[root2]=root1
this.rank[root1]+=this.rank[root2]
this.setCount--
}
}
getCount(){
return this.setCount
}
}
var findCircleNum = function (isConnected) {
let circleNum = isConnected.length;
const uf = new UnionFind(circleNum);
for (let index1 = 0; index1 < circleNum; index1++) {
for (let index2 = index1+1; index2 < circleNum; index2++) {
if(isConnected[index1][index2]==1){
uf.unite(index1,index2)
}
}
}
return uf.getCount()
};