[LeetCode] 547. Friend Circles

原创于 2017-09-05 13:45:47 发布 · 348 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

37 篇文章

订阅专栏

本文介绍了一种使用并查集算法解决朋友圈问题的方法。通过定义并实现特定的并查集操作，如查找根节点和合并集合，来计算给定矩阵中互为朋友的人群数量。该算法首先初始化每个元素为其自身的父节点，然后遍历矩阵，根据矩阵中的1将相关联的元素合并到同一集合中，最后统计剩余的独立集合数量即为朋友圈的数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

int findCircleNum(vector<vector<int>>& M)
{
    int n=M.size();
    vector<int> parents(n,0);
    for(int i=0;i<n;++i)
    {
        parents[i]=i;
    }
    function<int(int,vector<int>&)>finds=[&finds](int x,vector<int> &parents)->int{return parents[x]==x?x:parents[x]=finds(parents[x],parents);};
    int cnt=n;
    for(int i=0;i<n;++i)
    {
        for(int j=i+1;j<n;++j)
        {
            if(M[i][j]==1)
            {
                int ip=finds(i,parents);
                int jp=finds(j,parents);
                if(ip!=jp)parents[ip]=jp,cnt--;
            }
        }

    }
    return cnt;
}