LeetCode 684. 冗余连接 Redundant Connection(Java)

Posted 2021-05-21 爱若信若盼若

0684. 冗余连接 Redundant Connection(Medium)

##Union Find##

并查集

并查集可以维护无向图的联通分量

• 对边数组中每条边进行遍历
• 每条边的两个端点，查找其所属的联通分量
  • 若两端点所属联通分量不相同，代表两端点不连通，联通两端点后也不会形成环
  • 若两端点所属的联通分量相同，两端点联通，联通两端点后会行程环，该边就是形成环需要删去的最后的边
• 若所有边都不成环，则返回空数组

时间复杂度：$O(n)$
将并查集操作复杂度视作$O(1)$

class Solution {
    int[] p;

    public int find(int x) {
        if (x != p[x]) p[x] = find(p[x]);
        return p[x];
    }

    public int[] findRedundantConnection(int[][] edges) {
        int n = edges.length;
        p = new int[n + 1];
        
        for (int i = 0; i <= n; i ++) {
            p[i] = i;
        }

        for (int[] edge : edges) {
            int a = find(edge[0]);
            int b = find(edge[1]);
            if (a == b) return edge;
            else {
                p[b] = a;
            }

        return new int[0];

    }
}