首页 > 其他分享 >省份数量

省份数量

时间:2023-03-03 17:25:56浏览次数:53  
标签:int 城市 visit isConnected 相连 省份 数量

省份数量

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

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

解题思路

同岛屿数量类似,遍历每一个城市,使用广度优先搜素or深度优先搜索or并查集查找处于同一个省份的城市。

code 广度优先搜索

class Solution {
public:
    int findCircleNum(vector<vector<int>>& isConnected) {
        int ans = 0;
        int n = isConnected.size();
        
        vector<int> visit(n,0);

        for(int i = 0;i < n;i ++)
        {
            if(!visit[i])
            {
                ans ++;
                queue<int> q;
                q.push(i);
                visit[i] = 1;
                while(!q.empty())
                {
                    int cur = q.front();
                    q.pop();
                    for(int j = 0;j < n;j++)
                    {
                        if(!visit[j] && isConnected[cur][j])
                        {
                            visit[j] = 1;
                            q.push(j);
                        }
                    }
                }
            }
        }

        return ans;
    }
};

标签:int,城市,visit,isConnected,相连,省份,数量
From: https://www.cnblogs.com/huangxk23/p/17176320.html

相关文章