[路飞]leetcode-547.省份数量

126 阅读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

解题:

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()
};