leetcode 684. 冗余连接（并查集）

最新推荐文章于 2023-05-06 19:56:55 发布

翻译最新推荐文章于 2023-05-06 19:56:55 发布 · 231 阅读

文章标签：

#leetcode 684. 冗余连接（并查集）

并查集专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一种使用并查集数据结构解决图中冗余连接问题的方法。通过定义并实现UnionFind类，该类提供了查找和并集操作，用于检测图中是否存在形成环的边。在遍历所有边的过程中，若发现某条边连接的两个顶点已处于同一集合，则认为这条边是冗余的。最后返回最后一条形成环的边作为冗余连接。

在这里插入图片描述

	class unionfind{
public:
    vector<int>father;
    unionfind(const int &num){//num表示元素的个数
        for(int i = 0; i < num; i++){
            father.push_back(i);//箭头指向自己
        }
    }
    int find(const int &n){
        //递归
        if(father[n] == n)
            return n;
        father[n] = find(father[n]);//路径压缩版本
        return father[n];
    }
    bool myunion(const int &a,const int &b){
        if(father[a]==b||father[b]==a)return true;//判断是否是一个集合元素
        int fa = find(a);
        int fb = find(b);
        bool res=fa==fb;
        father[fb] = fa;
        return res;
    }
};

class Solution {
public:
    vector<int> findRedundantConnection(vector<vector<int>>& edges) {
       int n=edges.size();
       vector<int>ans(2,0);
       unionfind F(n+1);
       for(int i=0;i<n;i++)
       {
         int a=edges[i][0];
         int b=edges[i][1];
         bool isok=F.myunion(a,b);
         if(isok)//如果是一个集合的元素就更新答案，因为题目要求是最后一个；
         {
           ans[0]=a;
           ans[1]=b;
         }
         
       }
      return ans;
    }
};