【力扣-困难】685. 冗余连接 II

在本问题中，有根树指满足以下条件的有向图。该树只有一个根节点，所有其他节点都是该根节点的后继。每一个节点只有一个父节点，除了根节点没有父节点。
输入一个有向图，该图由一个有着N个节点 (节点值不重复1, 2, ..., N) 的树及一条附加的边构成。附加的边的两个顶点包含在1到N中间，这条附加的边不属于树中已存在的边。
结果图是一个以边组成的二维数组。 每一个边 的元素是一对 [u, v]，用以表示有向图中连接顶点 u 和顶点 v 的边，其中 u 是 v 的一个父节点。
返回一条能删除的边，使得剩下的图是有N个节点的有根树。若有多个答案，返回最后出现在给定二维数组的答案。

示例 1:

输入: [[1,2], [1,3], [2,3]]
输出: [2,3]
解释: 给定的有向图如下:

示例 2:

输入: [[1,2], [2,3], [3,4], [4,1], [1,5]]
输出: [4,1]
解释: 给定的有向图如下:

注意:

• 二维数组大小的在3到1000范围内。
• 二维数组中的每个整数在1到N之间，其中 N 是二维数组的大小。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/redundant-connection-ii

【并查集】

在一棵树中，边的数量比节点的数量少 1。如果一棵树有 N 个节点，则这棵树有 N-1 条边。
这道题中的图在树的基础上多了一条附加的边，因此边的数量也是 N。
树中的每个节点都有一个父节点，除了根节点没有父节点。在多了一条附加的边之后，可能有以下两种情况：

• 附加的边指向根节点，则包括根节点在内的每个节点都有一个父节点，此时图中一定有环路。
• 附加的边指向非根节点，则恰好有一个节点（即被附加的边指向的节点）有两个父节点，此时图中可能有环路也可能没有环路。

作者：LeetCode-Solution
链接：https://leetcode-cn.com/problems/redundant-connection-ii/solution/rong-yu-lian-jie-ii-by-leetcode-solution/

int[] unionFind;
    public int[] findRedundantDirectedConnection(int[][] edges) {
        int nodeCount = edges.length;
        unionFind = new int[nodeCount+1];
        int[] parent = new int[nodeCount+1];
        //初始化
        for(int i = 1; i <= nodeCount; i++){
            unionFind[i] = i;
            parent[i] = i;
        }
        int twoParent = -1;
        int circle = -1;
        for (int i = 0; i < edges.length; i++) {
            int fromNode = edges[i][0], toNode = edges[i][1];
            //已有父节点
            if(parent[toNode] != toNode){
                twoParent = i;
            }else {
                parent[toNode] = fromNode;
                //fromNode和toNode之间存在环
                if (find(fromNode) == find(toNode)){
                    circle = i;
                }else {
                    union(fromNode, toNode);
                }
            }
        }
        //如果父节点+环路同时出现，应破坏前一个父节点，这是父节点也在环路
        if (twoParent >= 0 && circle >= 0){
            return new int[]{parent[edges[twoParent][1]], edges[twoParent][1]};
        }
        if (twoParent >= 0){
            return edges[twoParent];
        }else {
            return edges[circle];
        }
    }

    //查找index能到达的最远结点
    private int find(int node){
        if (unionFind[node] != node){
            unionFind[node] = find(unionFind[node]);
        }
        return unionFind[node];
    }

    //合并，根据index1到index2的连通，建立之间的可到达路径
    private void union(int fromNode, int toNode){
        unionFind[find(fromNode)] = find(toNode);
    }