【路飞】算法与数据结构-省份数量

不管全世界所有人怎么说，我都认为自己的感受才是正确的。无论别人怎么看，我绝不打乱自己的节奏。喜欢的事自然可以坚持，不喜欢的怎么也长久不了。

LeetCode：原题地址

题目要求

有 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]

思路

• 创建深度优先遍历函数：从入口开始，将相邻的城市全部进行访问
• 遍历城市，遇到没有访问过的，访问到底，访问完毕后，省份数量+1

const findCircleNum = isConnected => {
    // 行、列
    const [m, n] = [isConnected.length, isConnected[0].length];
    let res = 0;
    const visited = new Array(m).fill(0);
    // 深度优先遍历
    const dfs = i => {
        // 标记访问
        visited[i] = 1;
        for (let j = 0; j < n; j++) {
            // 查找与i相连，且没有访问过的，继续dfs
            if (isConnected[i][j] && !visited[j]) dfs(j);
        }
    };
    for (let i = 0; i < m; i++) {
        if (!visited[i]) {
            // 遇到没有访问过的，访问到底
            // 省份+1
            dfs(i);
            res++;
        }
    }
    return res;
};