547. 省份数量

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

来源:力扣(LeetCode)
链接:leetcode-cn.com/problems/nu…

实现

void dfs(int **isConnected, int i, int *visited, int isConnectedSize)
{
    for (int j = 0; j < isConnectedSize; j++) {
        if (isConnected[i][j] == 1 && visited[j] == 0) {    // j没有访问过,并且与i相连
            visited[j] = 1;
            dfs(isConnected, j, visited, isConnectedSize);
        }
    }
}

int findCircleNum(int **isConnected, int isConnectedSize, int *isConnectedColSize)
{
    if (isConnected == NULL || isConnectedSize == 0) {
        return 0;
    }
    int sum = 0;
    int visited[isConnectedSize];
    memset(visited, 0, sizeof(visited));

    // 遍历每一个城市,如果没有访问过,就深度搜索访问他,查看与他相连的城市
    for (int i = 0; i < isConnectedSize; i++) {
        if (visited[i] == 0) {  // 未访问
            dfs(isConnected, i, visited, isConnectedSize);
            sum++;
        }
    }
    return sum;
}