Java实现 LeetCode 685 冗余连接 II（并查集+有向图）

685. 冗余连接 II

在本问题中，有根树指满足以下条件的有向图。该树只有一个根节点，所有其他节点都是该根节点的后继。每一个节点只有一个父节点，除了根节点没有父节点。

输入一个有向图，该图由一个有着N个节点 (节点值不重复1, 2, …, N) 的树及一条附加的边构成。附加的边的两个顶点包含在1到N中间，这条附加的边不属于树中已存在的边。

结果图是一个以边组成的二维数组。 每一个边 的元素是一对 [u, v]，用以表示有向图中连接顶点 u and v和顶点的边，其中父节点u是子节点v的一个父节点。

返回一条能删除的边，使得剩下的图是有N个节点的有根树。若有多个答案，返回最后出现在给定二维数组的答案。

示例 1:

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

示例 2:

输入: [[1,2], [2,3], [3,4], [4,1], [1,5]]
输出: [4,1]
解释: 给定的有向图如下:
5 <- 1 -> 2
     ^    |
     |    v
     4 <- 3

注意:

二维数组大小的在3到1000范围内。

二维数组中的每个整数在1到N之间，其中 N 是二维数组的大小。

class Solution {
      public int[] findRedundantDirectedConnection(int[][] edges) {
        int[] path = new int[edges.length+1];
        Arrays.fill(path, -1);
        int path2 = -1;
        for(int i = 0; i < edges.length; i++) {
            if (path[edges[i][1]] != -1) {
                path2 = i;
            } else {
                path[edges[i][1]] = i;
            }
        }
        //边中出现了入度为2的点
        if (path2 != -1) {
            int node = edges[path2][1];
            int firstEdge = path[node];
            int temp = firstEdge;
            while (temp != -1) {
                int index = edges[temp][0];
                temp = path[index];
                if (index == node) {
                    //找到第一条边引发的环
                    return edges[firstEdge];
                }
            }
            return edges[path2];//前一条边无环则直接删除后一条边
        }
        //所有点的入度都为1，必有环，要删除环中最后输入的边
        boolean[] visited = new boolean[edges.length+1];
        Arrays.fill(visited, false);
        int index = 1;
        while (!visited[index]) {
            visited[index] = true;
            int firstEdge = path[index];
            index = edges[firstEdge][0];
        }
        int result = path[index];
        int temp = edges[result][0];
        int edgeNum = 0;
        while (temp != index) {
            edgeNum = path[temp];
            result = Math.max(result, edgeNum);
            temp = edges[edgeNum][0];
        }
        return edges[result];
    }
}