并查集例题之冗余连接

一、题目描述

二、思路

题目要求构成一棵树，那么我们应该使图中的环消失，就应该随意去掉环中的一条边。然而题目又要求去掉这些边中在edges数组中最后出现的一条边，那么该如何寻找这最后一条边呢？请看解题方法。

三、解题方法

既然图中存在一个环，那么是不是将最后一条边加入的时候它才形成了环。我们就可以从前向后遍历每一条边（优先让前面的边连上），边的两个节点如果不在同一个集合，就加入集合（即：同一个根节点）。如果两个节点已经出现在了集合中，那么再次连接他们是不是就形成了一个环，这两个节点所形成的边是不是就是我们所需要的那最后一条边。如图所示：

四、代码

//去掉成环的那一条边：只有思路不会下手写代码
class Solution {
public:
    vector<int> f;
    int find(int x)
    {
        if(f[x]==x) return x;
        else return find(f[x]);
    }
    void merge(int x,int y)
    {
        int fx = find(x);
        int fy = find(y);
        //因为主函数中已经判断x和y的根节点是否为同一个，所以在此就可以不用判断
        f[fx] = fy;
    }

    vector<int> findRedundantConnection(vector<vector<int>>& edges) {
        vector<int> v;
        int m=edges.size();
        int n=edges[0].size();
        f.resize(m*n);
        for(int i=0;i<m;i++)
        {
            for(int j=0;j<n;j++)
            {
                f[i*n+j]=i*n+j;
            }
        }
        for(int i=0;i<m;i++)
        {
           
            if(find(edges[i][0])==find(edges[i][1]))
            {
                v.push_back(edges[i][0]);
                v.push_back(edges[i][1]);
            }
            else
            {
                merge(edges[i][0],edges[i][1]);
            }
        
        }
        return v;
    }
};