有 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/number-of-provinces
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路及方法
运用BFS的方法,先找到一条路径的所有连接城市,然后+1,接着找下一条。
首先审题,参数矩阵是一个方阵且是对称矩阵,对角线都是1代表自身相连。所以我们从0开始遍历,代表第一个城市,先判断是否访问,未被访问就入队,然后寻找所有直接和间接相连接的城市,即矩阵值为1且未访问的城市,然后入队。
class Solution {
public int findCircleNum(int[][] isConnected) {
int n = isConnected.length;
int res = 0;
// 记录已访问城市
List<Integer> visited = new ArrayList<>();
// 记录路径
Queue<Integer> queue = new LinkedList<>();
for (int i = 0; i < n; i++) {
// 未访问城市入队
if (!visited.contains(i)) {
queue.offer(i);
while (!queue.isEmpty()) {
int cur = queue.poll();
// 当前城市已访问
visited.add(cur);
// 寻找连接城市
for (int j = 0; j < n; j++) {
if (isConnected[cur][j] == 1 && !visited.contains(j)) {
// 连接城市入队
queue.offer(j);
}
}
}
res++;
}
}
return res;
}
}
结果如下:
辣鸡
网友评论