刷题day55：省份数量

最新推荐文章于 2024-10-03 13:36:06 发布

Kbbl_dh

最新推荐文章于 2024-10-03 13:36:06 发布

阅读量74

点赞数

文章标签：算法动态规划

本文链接：https://blog.csdn.net/Kbbl_dh/article/details/130201681

版权

该文介绍了三种方法来计算给定城市矩阵中省份的数量。首先，使用深度优先搜索（DFS）遍历矩阵，当找到未访问过的城市时递归地标记其连接城市。其次，采用广度优先搜索（BFS）进行遍历。最后，利用并查集（Union-Find）数据结构初始化并合并城市，找出连通组件。所有方法的目标是确定直接或间接相连的城市组，即省份。

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

题意描述：

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

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

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

返回矩阵中省份的数量。

思路一：深度优先算法：

完整C++代码如下：

class Solution {
public:
    void dfs(vector<vector<int>>& isConnected, vector<int>& visited, int i) {
        for (int j = 0; j < isConnected.size(); j++) {
            if (isConnected[i][j] == 1 && !visited[j]) {
                visited[j] = 1;
                dfs(isConnected, visited, j);
            }
        }
    }
    int findCircleNum(vector<vector<int>>& isConnected) {
        vector<int> visited(isConnected.size());
        int provinces = 0;
        for(int i = 0; i < isConnected.size(); i++){
            if(!visited[i]){
                dfs(isConnected, visited, i);
                provinces++;
            }
        }
        return provinces;
    }
};

思路二：广度优先算法

完整C++代码如下：

class Solution {
public:
    int findCircleNum(vector<vector<int>>& isConnected) {
        vector<int> visited(isConnected.size());
        int provinces = 0;
        queue<int> que;
        for(int i = 0; i < isConnected.size(); i++){
            if(!visited[i]){
                que.emplace(i);
                while(!que.empty()){
                    int j = que.front();
                    que.pop();
                    visited[j] = 1;
                    for(int k = 0; k < isConnected.size(); k++){
                        if(isConnected[j][k] == 1 && !visited[k]){
                            que.emplace(k);
                        }
                    }
                }
                provinces++;
            }
        }
        return provinces;
    }
};

思路三：并查集

分为两个部分：初始化；合并

完整C++代码如下：

class Solution {
public:
    int find(vector<int>& parent, int x){
        if(parent[x] != x){
            parent[x] = find(parent, parent[x]);
        }
        return parent[x];
    }
    int findCircleNum(vector<vector<int>>& isConnected) {
        int cities = isConnected.size();
        vector<int> parent(cities);

        // 初始化部分
        for(int i = 0; i < cities; i++){
            parent[i] = i;
        }
        for(int i = 0; i < cities; i++){
            for(int j = 0; j < cities; j++){
                if(isConnected[i][j] == 1){
                    int index1 = find(parent, i);
                    int index2 = find(parent, j);
                    if(index1 != index2){
                        parent[index1] = index2; // 合并部分
                    }
                }
            }
        }
        int res = 0;
        for(int i = 0; i < cities; i++){
            if(parent[i] == i){
                res++;
            }
        }
        return res;
    }
};