LeetcodeLCR 116. 省份数量

题目

原题链接

LCR 116. 省份数量

思路

利用并查集的思想，将连接的诚实放在一个集合当中，最后遍历并查集数组判断有几颗树

• 初始化一个并查集；
• 将连通的城市合并；
• 统计并查集中树的个数；

C++代码

class Solution 
{
public:
    int findCircleNum(vector<vector<int>>& isConnected) 
    {
        // 初始化并查集
        vector<int> ufs(isConnected.size(), -1); 
    
        auto findRoot = [&ufs](int x)
        {
            int parent = x;
            while (ufs[parent] >= 0) 
                parent = ufs[parent];

            return parent;
        };

        for(int i = 0; i < isConnected.size(); i++)
            for(int j = 0; j < isConnected[0].size(); j++)
            {
                // 是否连通
                if(isConnected[i][j] == 1)
                {
                    int root1 = findRoot(i);
                    int root2 = findRoot(j);
                    // 合并
                    if (root1 != root2) 
                    {
                        ufs[root1] += ufs[root2];
                        ufs[root2] = root1; 
                    }
                }
            }

        // 统计并查集中集合的个数
        int n = 0;
        for (const int& e : ufs) 
        {
            if (e < 0)
                n++;
        }
        return n;
    }
};