省份数量、深度优先搜索、广度优先搜索、并查集解法

最新推荐文章于 2022-08-31 22:59:32 发布

爱学习的羽

最新推荐文章于 2022-08-31 22:59:32 发布

阅读量211

点赞数

分类专栏： Coding habits

本文链接：https://blog.csdn.net/qq_37437892/article/details/114626542

版权

Coding habits 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

547. 省份数量

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

class Solution {
public:
    void dfs(vector<vector<int>>& isConnected,vector<bool>&visit,int province,int i){
        // 标记已经访问
        visit[i] = true;
        for(int j=0;j<province;++j){
            if(!visit[j]&&isConnected[i][j]==1)
            dfs(isConnected,visit,province,j);
        }
    }

// 连通图的遍历
    int findCircleNum(vector<vector<int>>& isConnected) {
        int count=0;
        vector<bool> visit(isConnected.size(),false);
        int province= isConnected.size();
        for(int i=0;i<province;++i){
            if(!visit[i]){
                dfs(isConnected,visit,province,i);
                count++;
            }
        }
        return count;
        

        // 并查集解法
        // int count =0;
        // int tempi,tempj;
        // vector<int> gen(isConnected.size(),-1);
        // for(int i=0;i<isConnected.size();++i){
        //     for(int j =i+1;j<isConnected[0].size();++j){
        //         if(isConnected[i][j]==1){
        //             tempi = find(gen,i);
        //             tempj = find(gen ,j);
        //             if (tempi!=tempj){
        //                 lian(gen,tempi,tempj);
        //             }
        //         } 

        //     }
        // }
        // for(int i=0;i<isConnected.size();++i){
        //     if (gen[i]==-1) count++;
        // }
        // return count;
    }


    // // 并查集

    // int find(vector<int>&gen,int i){
    //     while(gen[i]!=-1){
    //         i=gen[i];
    //     }
    //     return i;
    // }

    // void lian(vector<int>&gen,int i,int j){
    //     // 小的节点作为根
    //     if(i<j){
    //         gen[j] = i;
    //     }
    //     else gen[i] = j;
    // }

  
};

class Solution(object):
    def findCircleNum(self, isConnected):
        """
        :type isConnected: List[List[int]]
        :rtype: int
        """
        # 广度优先搜索
        province = len(isConnected)
        visit = [False for i in range(province)]
        # print(visit)
        que = []
        count =0
        # count 记录城市群的个数
        for i in range(province):
            # 如果该城市没有访问
            if visit[i]==False:
                # 把该城市加入到列表中
                que.append(i)
                while len(que)!=0:
                    # 访问该城市，并把该城市从列表中删除
                    tempi = que.pop()
                    visit[tempi]= True
                    for j in range(province):
                        # 把与当前访问并且连通的城市加入到队列中，且此城市还没有被访问
                        if isConnected[tempi][j]==1 and visit[j]==False:
                            que.append(j)
                
                count+=1
        return count