省份数量

最新推荐文章于 2021-08-18 23:35:22 发布

KobeSacre

最新推荐文章于 2021-08-18 23:35:22 发布

阅读量172

点赞数

分类专栏： leetcode-bfs leetcode-dfs

本文链接：https://blog.csdn.net/KobeSacre/article/details/112303810

版权

leetcode-dfs 同时被 2 个专栏收录

31 篇文章 1 订阅

订阅专栏

leetcode-bfs

17 篇文章 1 订阅

订阅专栏

这篇博客介绍了一种解决寻找矩阵中省份数量的方法，通过深度优先搜索（DFS）和广度优先搜索（BFS）算法实现。给定一个表示城市连接的矩阵，省份是由直接或间接相连的城市组成，互不相连的城市不在同一省份。示例展示了如何应用这两种算法，最终返回矩阵中省份的个数。

摘要由CSDN通过智能技术生成

省份数量

题目：
有 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

解题思路：本题可以用dfs或者bfs来进行解答，遍历数组，当遇到没有到达过的城市时，找到该城市可以达到并且之前没有达到过的城市即可

DFS:

class Solution {
    public int findCircleNum(int[][] isConnected) {
        int n = isConnected.length;
        
        boolean book[] = new boolean[n];
        int ans = 0;
        for(int i = 0; i < n; i++) {
            if(!book[i]) {
                dfs(i, isConnected, book);
                ans++;
            }
        }
        
        return ans;
    }
    
    private void dfs(int cur, int[][] isConnected, boolean[] book) {
        
        book[cur] = true;
        
        for(int i = 0; i < isConnected.length; i++) {
            if(!book[i] && isConnected[cur][i] == 1) {
                dfs(i, isConnected, book);
            }
        }
        
    }
}

BFS

class Solution {
    public int findCircleNum(int[][] isConnected) {
        int n = isConnected.length;
        
        boolean book[] = new boolean[n];
        
        Queue<Integer> queue = new LinkedList();
        
        int ans = 0;
        
        for(int i = 0; i < n; i++) {
            if(!book[i]) {
                queue.offer(i);
                book[i] = true;
                while(!queue.isEmpty()) {
                    int cur = queue.poll();
                    for(int j = 0; j < n; j++) {
                        if(!book[j] && isConnected[cur][j] == 1) {
                            queue.offer(j);
                            book[j] = true;
                        }
                    }
                }
                
                ans++;
            }
        }
        
        return ans;
    }
}