Leetcode---685. 冗余连接 II---每日一题---并查集(有向环路问题)

685. 冗余连接 II

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

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

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

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

示例 1:

示例 2:

注意:

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

分析

1. 若存在入度为2的点时(即示例一)那么环必定与此点相关，尝试删除两条边中的一条判断是否解除环，若可以则返回该边。
2. 若不存在入度为2的点时(即示例二)，那么必定存在回路，从最后一条边开始找入度为1的点(即除去根节点)，逐一删去看是否可以接触环，若可以，则返回该边。

实现代码

class UnionFind{
private:
    vector<int> parent;

public:
    UnionFind(int n){
        parent = vector<int>(n + 1);
        for(int i = 1; i <= n; i++) parent[i] = i;
    }

    int find(int num){
        if(parent[num] == num) return num;
        return find(parent[num]);
    }

    bool Union(int a, int b){
        int pa = find(a), pb = find(b);

        if(pa == pb) return false;

        parent[b] = pa;
        return true;
    }
};

class Solution {
private:
    bool judgeCircle(vector<vector<int>> edges, int n, int delete_edge){
        auto uf = UnionFind(n);
        for(int i = 0; i < n; i++){
            // 去掉删除边
            if(i == delete_edge) continue;
            // 若是存在环，必定有回路形成
            if(!uf.Union(edges[i][0], edges[i][1]))  return true;

        }
        return false;
    }


public:
    vector<int> findRedundantDirectedConnection(vector<vector<int>>& edges) {
        int target = 1;
        int n = edges.size();
        vector<int> in_degree = vector<int>(n + 1, 0);

        for(auto edge: edges){
            int u = edge[0], v = edge[1];
            if(++in_degree[v] == 2) target = 2; // 若存在入度为2的点，target = 2
        }

        for(int i = n - 1; i >= 0; i--){
            int v = edges[i][1];
            // 入度为2时target = 2，此处必存在环
            // 无入度2点时，存在回路，此处target = 1，对所有入度为1的点进行检测
            if(in_degree[v] == target){
                if(!judgeCircle(edges, n, i)) return edges[i];
            }
        }
        return vector<int>();
    }
};