标签：dsu cur parent int res 查集 c++ vector leetcode

#include <iostream>
#include <vector>

using namespace std;

class DSU{
    public:
    vector<int>parent;
    DSU(int n)
    {
        parent = vector<int>(n);
        for(int i = 0; i< n; ++i)
        {
            parent[i] = i;
        }        
    }
    int Find(int x)
    {
        if(parent[x] != x)parent[x] = Find(parent[x]);
        return parent[x];
    }

    void Union(int x, int y)
    {
        parent[Find(x)] = Find(y);
    }
};

int findCircleNums(const vector<vector<int>>& M)
{
    DSU dsu(M.size());
    for(int i = 0; i < M.size(); ++i)
    {
        for(int j = 0; j < M[0].size(); ++j)
        {
            if(M[i][j] == 1)dsu.Union(i,j);
        }        
    }
    int res = 0;
    for(int i = 0; i < M.size(); ++i)
    {
        if(dsu.Find(i) == i) ++res;
    }
    return res;
}

vector<vector<int>>dirs{{0,1},{0,-1},{1,0},{-1,0}};

vector<int> numIslands(int m, int n, const vector<vector<int>>& position)
{
    DSU dsu(m * n);
    vector<vector<bool>>island(m,vector<bool>(n,false));
    vector<int>res;
    int cnt = 0;
    for(auto cur:position)
    {
        if(island[cur[0]][cur[1]])
        {
            res.push_back(cnt);
            continue;
        }
        island[cur[0]][cur[1]] = true;
        ++cnt;
        for(auto dir:dirs)
        {
            int x = cur[0] + dir[0],y = cur[1] + dir[1];
            if(x < 0 || x >= m || y < 0 || y >= n || island[x][y] == false)continue;
            int compoent1 = dsu.Find(cur[0] * n + cur[1]);
            int compoent2 = dsu.Find(x * n + y);
            if(compoent1 != compoent2)
            {
                dsu.Union(compoent1,compoent2);
                --cnt;
            }
        }
        res.push_back(cnt);
    }   
    return res;
}

int main()
{

    //LeetCode547
    vector<vector<int>>nums{{1,1,0},{1,1,0},{0,0,1}};
    cout << findCircleNums(nums) << endl;

    //LeetCode305
    vector<vector<int>>islands{{0,0},{0,1},{1,2},{2,1}};
    auto res = numIslands(3,3,islands);
    for(int num:res)
    {
        cout << num << " ";
    }
    cout << endl;
    return 0;
}

标签：dsu,cur,parent,int,res,查集,c++,vector,leetcode
来源： https://www.cnblogs.com/fourmi/p/16219315.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。