684. 冗余连接

题目描述：在本问题中, 树指的是一个连通且无环的无向图。
输入一个图，该图由一个有着N个节点 (节点值不重复1, 2, …, N) 的树及一条附加的边构成。附加的边的两个顶点包含在1到N中间，这条附加的边不属于树中已存在的边。
结果图是一个以边组成的二维数组。每一个边的元素是一对[u, v] ，满足 u < v，表示连接顶点u 和v的无向图的边。
返回一条可以删去的边，使得结果图是一个有着N个节点的树。如果有多个答案，则返回二维数组中最后出现的边。答案边 [u, v] 应满足相同的格式 u < v。
解题思路：无向无环图多增加了一条边，那必然会形成环路，利用并查集，依次遍历题目给出的边，如果当前两个节点所属连通域不同就合并当前两个集合，如果他们已经联通，因为当前这条新增的边导致形成环路，因此当前这条边作为候选结果，如果之后还有这种情况，更新返回结果，代码如下：

class Solution:
    def findRedundantConnection(self, edges: List[List[int]]) -> List[int]:
        n = len(edges)
        father = list(range(n+1))
        res = []
        
        def find(v):
            root = v
            while(root != father[root]):
                root = father[root]
            while(v != root):
                origin_root = father[v]
                father[v] = root
                v = origin_root
            return root
        def merge(a, b):
            root_a, root_b = find(a), find(b)
            if root_a != root_b:
                father[root_b] = root_a
        for e in edges:
            if find(e[0]) == find(e[1]):
                res = e
            else:
                merge(e[0], e[1])
        return res