[路飞]_js算法:leetcode 547-省份数量

126 阅读1分钟

leetcode 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

思路一: 并查集 储存到map对象中

/**
 * @param {number[][]} isConnected
 * @return {number}
 */
var findCircleNum = function(isConnected) {
  let unionSize=new unionSizes(isConnected.length);
  let size=isConnected.length;
  for(let i=0;i<size;i++){
    for(let j=0;j<size;j++){
      if(isConnected[i][j]==1)unionSize.merge(i,j);
    }
  }
  let map=new Map
  for(let item of unionSize.node){
    // if(!map.has(item))
    let tem=unionSize.find(item)
    map.set(tem,(map.get(tem)||0)+1)

  }
  return [...map.keys()].length

};

思路二: 并查集 找到集合节点是他自己本身的

var findCircleNum = function(isConnected) {
  let unionSize=new unionSizes(isConnected.length);
  let size=isConnected.length;
  for(let i=0;i<size;i++){
    for(let j=0;j<size;j++){
      if(isConnected[i][j]==1)unionSize.merge(i,j);
    }
  }
  let res=0;
  for(let item of unionSize.node.entries()){
   if(unionSize.find(item[1])==item[0])res++
  }
  return res

};
// 并查集
class unionSizes{
  constructor(n){
    this.node=new Array(n).fill(0).map((item,index)=>index);
  }
  find(x){
  return this.node[x]=(x==this.node[x]?x:this.find(this.node[x]))
  }
  merge(x,y){
    let fa=this.find(x),fb=this.find(y);
   this.node[fa]=fb;
  }
}