[路飞]_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

题目解释

并：合并
查：查找
集：集合
例如：一个朋友圈，a和b是朋友，b和c是朋友，c和d是朋友，那么a和d也是朋友，因为abcd是在一个朋友圈集合内。

代码

var findCircleNum = function(isConnected){
    let len = isConnected.length;
    let list = new Array(len).fill(0).map((value, index) => index);
    for(let i = 0; i < len; i++){
        for(let j = 0; j < len; j++){
            if(isConnected[i][j] == 1){
                merge(i, j);
            }
        }
    }
    // 连通分量的特点是当前值等于自身
    let ans = 0;
    for(let i = 0;  i < len; i++){
        if(list[i] == i) ans += 1;
    }
    return ans;

    // 查找
    function find(x){
        // 如果当前节点就是自己的父节点，那么就返回当前节点
        if(list[x] == x) return x;
        // 递归查找当前节点的父节点
        const root = find(list[x]);
        list[x] = root;
        return root;
    }
    // 合并
    function merge(i, j){
        let root1 = find(i);
        let root2 = find(j);
        // 分别查找当前i和j节点的父节点
        // 如果根节点相同，说明本身就在同一个集合内
        if(root1 == root2) return;
        // 否者把节点root2挂到root1上
        list[root1] = root2;
    }
}