LeetCode 684. Redundant Connection [Medium]

原题地址

题目内容

题目分析

题目的意思大致为，给了一个无向图，让我们删掉组成环的最后一条边。一开始想的做法是想采用拓扑排序来判断是否由环，如果有再删去多余的边。后面看了leetcode的discuss，发现他用了一个较为简单的方法，union find。建立一个长数组，来存储每个节点，初始化为-1，每加入一条新的边（u，v)，都通过find()函数来查找u之前是否有一条到V的路，如果有，那么这条边就是多余的，如果没有就添加到root数组里面,root[u] = v,就代表有一条u->v的路径。

代码实现

class Solution {
public:
    vector<int> findRedundantConnection(vector<vector<int>>& edges) {
        vector<int> root(20001,-1);
        for(auto i:edges){
            int a = find(root,i[0]);
            int b = find(root,i[1]);
            if(a == b){
                return i;
            }
            else{
                root[a] = b;
            }
        }

    }
    int find(vector<int>&root,int x){
        while(root[x] != -1){
            x = root[x];
        }
        return x;
    }
};