leetcode 547. 朋友圈 medium

最新推荐文章于 2022-07-06 12:17:19 发布

原创最新推荐文章于 2022-07-06 12:17:19 发布 · 454 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#leetcode #算法

leetcode 同时被 2 个专栏收录

218 篇文章

订阅专栏

图和并查集

10 篇文章

订阅专栏

本文详细解析了LeetCode上的经典题目“朋友圈”(547)的算法解决方案，介绍了如何使用深度优先搜索（DFS）、广度优先搜索（BFS）以及并查集（Union-Find）算法来解决此问题，旨在帮助读者理解社交网络中朋友圈的数学模型及其计算方法。

leetcode 547. 朋友圈 medium

题目描述：

班上有 N 名学生。其中有些人是朋友，有些则不是。他们的友谊具有是传递性。如果已知 A 是 B 的朋友，B 是 C 的朋友，那么我们可以认为 A 也是 C 的朋友。所谓的朋友圈，是指所有朋友的集合。

给定一个 N * N 的矩阵 M，表示班级中学生之间的朋友关系。如果M[i][j] = 1，表示已知第 i 个和 j 个学生互为朋友关系，否则为不知道。你必须输出所有学生中的已知的朋友圈总数。

示例 1:

输入:
[[1,1,0],
[1,1,0],
[0,0,1]]
输出: 2
说明：已知学生0和学生1互为朋友，他们在一个朋友圈。
第2个学生自己在一个朋友圈。所以返回2。

注意：

N 在[1,200]的范围内。
对于所有学生，有M[i][i] = 1。
如果有M[i][j] = 1，则有M[j][i] = 1。

解题思路：

方法1：跟leetcode 200 岛屿数量一样，可以用dfs or bfs

方法2：用并查集

代码：

class Solution {
public:
    int findCircleNum(vector<vector<int>>& isConnected) {

        UF uf(isConnected.size());

        // 因为是邻接矩阵，所以遍历半个矩阵就够了
        for (int i = 0; i< isConnected.size(); i++){
            for (int j = 0; j < i; j++){
                if (isConnected[i][j])
                    uf.union_(i, j);
            }
        }

        return uf.get_count();
    }

private:
    class UF{
    public:
        UF(int n): id(n,0), size(n, 1), count(n){
            for (int i = 0; i<n; i++)
                id[i] = i;
        }

        int find(int p){
            while (p != id[p]){
                id[p] = id[id[p]];
                p = id[p];
            }
            return p;
        }

        void union_(int p, int q){
            p = find(p);
            q = find(q);
            if (p != q){
                if (size[p] > size[q]){
                    id[q] = p;
                    size[p] += size[q];
                } else {
                    id[p] = q;
                    size[q] += size[p];
                }
                count--;
            }
        }

        int get_count(){
            return count;
        }

    private:
        vector<int> id;
        vector<int> size;
        int count;
    };



};




//dfs
class Solution {
public:
    int findCircleNum(vector<vector<int>>& vec) {
        if(vec.size()<=1)
            return vec.size();
        int m=vec.size();
        int res=0;
        vector<bool> visited(m,false);
        for(int i=0;i<m;++i){
            if(!visited[i]){
                dfs(i,vec,visited);
                ++res;
            }
        }
        
        return res;
    }
    
    void dfs(int cur,vector<vector<int>> &vec, vector<bool> &visited){
        visited[cur]=true;
        for(int i=0;i<vec[cur].size();++i){
            if(!visited[i] && vec[cur][i]==1)
                dfs(i,vec,visited);
        }
    }
    
};