leetcode684.冗余连接

 

 依旧是并查集问题，这道题目正好给定顶点数目和边的数目相等，只要找到其中的一条边删除将图转化为树就行，而这个多余的边起始就是并查集的添加过程中二者是同一个根（两个顶点早已经联通了），这时直接返回这条边就行

注意这题中的顶点编号从1开始，为保证一致性，father数组大小为n+1，0索引不使用

class Solution {
private:
    vector<int> father;

    void init() {
        for(int i=0;i<father.size();i++)
            father[i]=i;
    }

    int find(int v) {
        return v==father[v]?v:father[v]=find(father[v]);
    }

    bool isSame(int u,int v) {
        u=find(u);
        v=find(v);
        return u==v;
    }

    bool join(int u,int v) {
        u=find(u);
        v=find(v);
        if(u==v)
            return false;
        else{
            father[v]=u;
            return true;
        }
    }
public:
    vector<int> findRedundantConnection(vector<vector<int>>& edges) {
        int n=edges.size();
        this->father=vector<int>(n+1);
        this->init();

        for(int i=0;i<n;i++){
            int u=edges[i][0];
            int v=edges[i][1];
            if(!this->join(u,v))
                return {u,v};
        }
        return {};
    }
};