[LeetCode] 547. Friend Circles 朋友圈

      本题可以套用和 [LeetCode] 200. Number of Islands 岛屿的数量 与 [LeetCode 695] Max Area of Island 岛的最大面积一样的dfs模版，不同的是DFS的遍历路径不同，本题

采用的是类似十字遍历方式，具体见如下代码。

/*
 * @Descripttion: 
 * @version: 
 * @Author: wangxf
 * @Date: 2019-12-20 15:21:49
 * @LastEditors: Do not edit
 * @LastEditTime: 2020-04-12 23:25:24
 */
/*
 * @lc app=leetcode.cn id=547 lang=cpp
 *
 * [547] 朋友圈
 */

// @lc code=start
#include<bits/stdc++.h>
using namespace std;
class Solution {
public:
    int findCircleNum(vector<vector<int>>& M)
    {
        int res = 0;
        if(M.empty()||M.size()<=0) return 0;
        for (size_t i = 0; i < M.size(); i++)
            for (size_t j = 0; j < M[0].size(); j++)
            {
                /*
                if(i==j)
                {
                  //++res;
                  continue;
                }
                */ 
                if(M[i][j]==1)
                {
                    ++res;
                    dfs(M,i,j);
                }
            }
        return res;  
    }
    void dfs(vector<vector<int>>& M,int i,int j)
    {
        if(!(i>=0&&i<M.size()&&j>=0&&j<M[0].size()))
        {
            return;
        }
        if(M[i][j]==0)
        {
            return;
        }
        M[i][j]=0;
        //if(i==j) return;

        for(int r = 0;r<M.size();++r)
        {
            if(M[r][j]==1&&r!=i)
            {
                dfs(M,r,j);
            }
        }
        for(int c = 0;c<M[0].size();++c)
        {
            if(M[i][c]==1&&c!=j)
            {
                dfs(M,i,c);
            }
        }
        return;
    }
};
// @lc code=end

 

     本题同样可以并查集的方法解决。代码如下：

/*
 * @Descripttion: 
 * @version: 
 * @Author: wangxf
 * @Date: 2019-12-20 15:21:49
 * @LastEditors: Do not edit
 * @LastEditTime: 2020-04-14 19:49:27
 */
/*
 * @lc app=leetcode.cn id=547 lang=cpp
 *
 * [547] 朋友圈
 */

// @lc code=start
#include<bits/stdc++.h>
using namespace std;

//union find 算法
class UF
{
public:
    // 构造函数初始化
    UF(int n)
    {
        count = n;
        parent.resize(n);
        size.resize(n);
        for(int i = 0; i < n; i++)
        {
            parent[i] =i;
            size[i] = 1;
        }
    }
    // 将节点p 和 节点 q 连通
    void Union(int p, int q)
    {
        int rootP = find(p);
        int rootQ = find(q);
        if(rootP == rootQ)
            return;
        
        if(size[rootP] > size[rootQ])
        {
            parent[rootQ] = rootP;
            size[rootP] += size[rootQ];
        }
        else
        {
            parent[rootP] = rootQ;
            size[rootQ] += rootP;
        }
        count--;//连通个数 -1
    }
    //判断节点p 和 节点 q 是否连通
    bool connected(int p, int q)
    {
        int rootP = find(p);
        int rootQ = find(q);
        return rootP==rootQ;
    }
    //返回当前连通个数
    int count_num()
    {
        return count;
    }
private:
    int count;//连通的个数
    vector<int> parent;//保存节点的父节点
    vector<int> size;//保存节点所在连通分量的重量(节点数)
    
    int find(int x)// 寻找 x 节点的根点
    {
        while(parent[x] != x)
        {
            //查找根节点的同时，对树进行路径压缩，
            parent[x] = parent[parent[x]];
            x = parent[x];
        }
        return x;
    }
};

class Solution {
public:
    int findCircleNum(vector<vector<int>>& M) {
        // base case
        if(M.size()==0)
            return 0;
        
        const int m = M.size();//行数
        const int n = M[0].size();//列数
        //UF* uf = new UF(m);
        UF uf(m);
        for(int i = 0; i < m; i++)
        {
            for(int j = i+1; j < n; j++)//表示朋友圈的矩阵是对称的
            {
                if(M[i][j] == 1)
                {
                    uf.Union(i,j);
                }
            }
        }
        return uf.count_num();
    }
};

// @lc code=end

---------------------------- ------------------------------参考链接---------------------------------------------------------------------------

感谢：

通俗理解并查集：https://blog.csdn.net/niushuai666/article/details/6662911

使用并查集解决朋友圈问题详解： https://github.com/labuladong/fucking-algorithm/blob/master/%E7%AE%97%E6%B3%95%E6%80%9D%E7%BB%B4%E7%B3%BB%E5%88%97/UnionFind%E7%AE%97%E6%B3%95%E8%AF%A6%E8%A7%A3.md

并查集算法应用：https://github.com/labuladong/fucking-algorithm/blob/master/%E7%AE%97%E6%B3%95%E6%80%9D%E7%BB%B4%E7%B3%BB%E5%88%97/UnionFind%E7%AE%97%E6%B3%95%E5%BA%94%E7%94%A8.md