LeetCode 547.省份数量

LeetCode 547.省份数量

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

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

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

返回矩阵中 省份 的数量。

class Solution {
public:
    int findCircleNum(vector<vector<int>>& isConnected) {
        int n = isConnected.size();
        int nums=0;
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++)
            {
                if(isConnected[i][j] == 1)
                {
                    ++nums;
                    dfs(isConnected,i,n);
                }
            }
        }
        return nums;
    }
    
    void dfs(vector<vector<int>>& isConnected,int &i,int &n) {
        for(int k=0;k<n;k++)
        {
            if(isConnected[i][k]==1)	// 如果i和k相连
            {
                isConnected[k][k] = 0;	// 消除k与自己的连接
                isConnected[i][k] = 0;	// 消除i与k的连接
                isConnected[k][i] = 0;	// 消除k与i的连接
                dfs(isConnected,k,n);	// 深度遍历与k相连的城市
            }
        }
    }
};