LeetCode 547. 省份数量

Posted 2021-10-10 Blocking The Sky

有 n 个城市，其中一些彼此相连，另一些没有相连。如果城市 a 与城市 b 直接相连，且城市 b 与城市 c 直接相连，那么城市 a 与城市 c 间接相连。

省份 是一组直接或间接相连的城市，组内不含其他没有相连的城市。

给你一个 n x n 的矩阵 isConnected ，其中 isConnected[i][j] = 1 表示第 i 个城市和第 j 个城市直接相连，而 isConnected[i][j] = 0 表示二者不直接相连。

返回矩阵中 省份 的数量。

深度优先搜索

class Solution {
public:
    bool visited[210];
    void dfs(int v,vector<vector<int>>& isConnected){//从顶点v出发，深度优先遍历
        visited[v]=true;
        for(int i=0;i<isConnected.size();i++){
            if(!visited[i]&&isConnected[v][i]==1)
                dfs(i,isConnected);
        }
    }
    int findCircleNum(vector<vector<int>>& isConnected) {
        int vexnum;
        int sum=0;
        vexnum=isConnected.size();
        for(int i=0;i<vexnum;i++){
            visited[i]=false;
        }
        for(int i=0;i<vexnum;i++){
            if(!visited[i]){
                dfs(i,isConnected);
                sum++;
            }
        }
        return sum;
    }
};

广度优先搜索

class Solution {
public:
    bool visited[210];
    int vexnum;
    void bfs(int v,vector<vector<int>>& isConnected){//从顶点v出发，广度优先遍历
        queue<int> q;
        visited[v]=true;
        q.push(v);
        int vex;
        while(!q.empty()){
            vex=q.front();
            q.pop();
            for(int i=0;i<vexnum;i++){
                if(!visited[i]&&isConnected[vex][i]==1){
                    visited[i]=true;
                    q.push(i);
                }
            }
        }
    }
    int findCircleNum(vector<vector<int>>& isConnected) {
        int sum=0;
        vexnum=isConnected.size();
        for(int i=0;i<vexnum;i++){
            visited[i]=false;
        }
        for(int i=0;i<vexnum;i++){
            if(!visited[i]){
                bfs(i,isConnected);
                sum++;
            }
        }
        return sum;
    }
};