[路飞]_leetcode 547. 省份数量

354 阅读1分钟

题目来源: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

    }
}