力扣685、冗余连接2

题目

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

输入一个有向图，该图由一个有着 n 个节点（节点值不重复，从 1 到 n）的树及一条附加的有向边构成。附加的边包含在 1 到 n 中的两个不同顶点间，这条附加的边不属于树中已存在的边。

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

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

题解

• 采用并采集

class Solution {
public:

/*
    思路 ： 逐个删除边，判断剩下的边是不是构成一个有向树
    判断思路 ： 
            1、有且只有一个节点入度为0 -> 无环
            2、集合没有节点的入度为2
*/  
    #define maxn 1001
    #define swap(a, b) \
    (a) ^= (b), (b) ^= (a), (a) ^= (b)

    int father[maxn], siz[maxn];
    int rudu[maxn];
    int find(int a) {
        return ( a != father[a] ? (father[a] = find(father[a])) : father[a] );
    }

    int merge(int a, int b) {// a->b
        if (b != father[b]) return 1;
        int fa = find(a), fb = find(b);
        if (fa == fb) {
            return 1;
        }
        father[fb] = fa;
        siz[fb] += siz[fa];
        return 0;
    }
    vector<int> findRedundantDirectedConnection(vector<vector<int>>& edges) {
        for (int i = edges.size() - 1; i >= 0; --i) {
            int flag1 = 0, flag2 = 0;
            for (int j = 0; j < maxn; ++j) {
                father[j] = j;
                siz[j] = 1;
                rudu[j] = 0;
            }
            //printf("aaaa\n");
            for (int j = 0; j < edges.size(); ++j) {
                if (j == i) continue;
                int a = edges[j][0], b = edges[j][1]; // a->b
                if ( (flag1 = merge(a, b)) == 1) break;
            
                if (++rudu[b] > 1) {
                    flag2 = 1;
                    break;
                }
            }
            if (!flag1 & !flag2) return edges[i];
        }
        return edges[0];
    }
};