547. 朋友圈 Medium

Leetcode笔记目录

---

一、题目描述

班上有 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 。

二、解题过程

1.思想

本题是一个求解无向图中连通分量个数的问题，可以用dfs，bfs，并查集，这里采用两种递归实现dfs和并查集；

2.代码

不判断的dfs递归实现：

    int dfs(vector<vector<int>>& M,vector<bool>& v,int k){
        if(v[k]==true)return 0;
        v[k]=true;
        for(int i=0;i<M.size();i++){
            if(M[k][i]==1)
                dfs(M,v,i);
        }
        return 1;
    }
    int findCircleNum(vector<vector<int>>& M) {
        int num=0;
        vector<bool> visit(M.size(),false);
        for(int i=0;i<M.size();i++){
            num += dfs(M,visit,i);
        }
        return num;
    }
};

• 注意
  • 此时不像Leetcode695可以数组值代替判断是否访问过，新增一个辅助数组visit是图遍历算法的常规操作;

先判断的dfs递归实现：

    void dfs(vector<vector<int>>& M,vector<bool>& v,int k){
        v[k]=true;
        for(int i=0;i<M.size();i++){
            if(M[k][i]==1&&v[k]!=true)
                dfs(M,v,i);
        }
    }
    int findCircleNum(vector<vector<int>>& M) {
        int num=0;
        vector<bool> visit(M.size(),false);
        for(int i=0;i<M.size();i++){
            if(!visit[i]){
                dfs(M,visit,i);
                num++;
            }
        }
        return num;
    }
};

• 注意
  • 先判断的结构;

并查集实现：

    int findFather(vector<int>& father,int k){
        int a=k;
        while(k!=father[k])k=father[k];
        while(a!=father[a]){
            int z=a;
            a=father[a];
            father[z]=k;
        }
        return k;
    }
    void Union(vector<int>& father,int k,int j){
        int a=findFather(father,k);
        int b=findFather(father,j);
        if(a!=b)father[a]=b;
    }
    int findCircleNum(vector<vector<int>>& M) {
        vector<int> father(M.size());   //错误2
        int num=0;
        for(int i=0;i<M.size();i++) father[i]=i;
        for(int i=0;i<M.size();i++)
            for(int j=i+1;j<M[0].size();j++){
                if(M[i][j]==1)Union(father,i,j);
            }
        for(int i=0;i<M.size();i++)
            if(father[i]==i)num++;
        return num;
    }

• 注意
  • 此处并查集使用了压缩路径，但没有使用其他的比如平衡策略等加快并查集的方法，速度仍然可观;
  • vector若使用push_back赋值，则声明时不要加范围，若使用一般数组赋值，则声明时一定要加范围;

---

三、总结

标准的无向图中连通分量个数的问题。