刷题-力扣-547

547. 省份数量

题目链接

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/number-of-provinces/
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题目描述

有 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. 根据问题描述，可以使用广度优先搜索和深度优先搜索算法。
2. 下面代码使用广度优先搜索的思想。
3. 实现方式是递归。
4. 依次遍历和上一个结点连接的结点。
5. 遍历过的结点做标记（arr）。
6. 直到每个结点都遍历过为止。

代码

class Solution {
public:
    int arr[200];
    int sum = 0;        // record sum number

    void recursion(vector<vector<int>>& isConnected, int index) {
        int n = isConnected.size();
        for (int i = 0; i < n; i ++) {
            if (isConnected[index][i] == 1 && index != i && arr[i] == 0) {
                arr[i] = 1;
                recursion(isConnected, i);
            }
        }
    }
    int findCircleNum(vector<vector<int>>& isConnected) {
        int n = isConnected.size();
        for (int j = 0; j < n; j++) {
           if (arr[j] == 0) {
               arr[j] == 1;
               sum++;
               recursion(isConnected, j);
           }
        }
        return sum;
    }
};

用时81min（首先尝试使用循环实现，失败。改为递归实现）