剑指 Offer II 118. 多余的边（中等 并查集 图）

剑指 Offer II 118. 多余的边

树可以看成是一个连通且 无环 的 无向 图。

给定往一棵 n 个节点 (节点值 1～n) 的树中添加一条边后的图。添加的边的两个顶点包含在 1 到 n 中间，且这条附加的边不属于树中已存在的边。图的信息记录于长度为 n 的二维数组 edges ，edges[i] = [ai, bi] 表示图中在 ai 和 bi 之间存在一条边。

请找出一条可以删去的边，删除后可使得剩余部分是一个有着 n 个节点的树。如果有多个答案，则返回数组 edges 中最后出现的边。

示例 1：

输入: edges = [[1,2],[1,3],[2,3]]
输出: [2,3]
示例 2：

输入: edges = [[1,2],[2,3],[3,4],[1,4],[1,5]]
输出: [1,4]

提示:

n == edges.length
3 <= n <= 1000
edges[i].length == 2
1 <= ai < bi <= edges.length
ai != bi
edges 中无重复元素
给定的图是连通的

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/7LpjUW
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

分析

依然是并查集的问题，对于一棵树，随便多出一个边都会形成环，我们只要找到最后多出的这条形成环的边即可。我们知道并查集的子集是一棵树，子集中所有节点的根节点只有一个，我们以 [[1,2],[2,3],[3,4],[1,4],[1,5]] 为例，前三个数组形成的子集会有一个相同的跟，即 1、2、3、4 有一个相同的根，对第四个数组的元素进行判断，1 和 4 已经有相同的根了，如果 1 和 4 再相连就会成环，这可以作为判定条件，即 union 方法如果返回 false 就说明最后一个使数形成环的数组找到了即 [i, j] 。

题解（Java）

class Solution {
    public int[] findRedundantConnection(int[][] edges) {
        int maxVertex = 0;
        for (int[] edge : edges) {
            maxVertex = Math.max(maxVertex, edge[0]);
            maxVertex = Math.max(maxVertex, edge[1]);
        }

        int[] fathers = new int[maxVertex + 1];
        for (int i = 1; i <= maxVertex; i++) {
            fathers[i] = i;
        }

        for (int[] edge : edges) {
            if (!union(fathers, edge[0], edge[1])) {
                return edge;
            }
        }

        return new int[2];
    }

    private int findFather(int[] fathers, int i) {
        if (fathers[i] != i) {
            fathers[i] = findFather(fathers, fathers[i]);
        }

        return fathers[i];
    }

    private boolean union(int[] fathers, int i, int j) {
        int fatherOfI = findFather(fathers, i);
        int fatherOfJ = findFather(fathers, j);
        if (fatherOfI != fatherOfJ) {
            fathers[fatherOfI] = fatherOfJ;
            return true;
        }

        return false;
    }
}