0684. 冗余连接 Redundant Connection(Medium)
##Union Find##
并查集
并查集可以维护无向图的联通分量
- 对边数组中每条边进行遍历
- 每条边的两个端点,查找其所属的联通分量
- 若两端点所属联通分量不相同,代表两端点不连通,联通两端点后也不会形成环
- 若两端点所属的联通分量相同,两端点联通,联通两端点后会行程环,该边就是形成环需要删去的最后的边
- 若所有边都不成环,则返回空数组
时间复杂度:
O
(
n
)
O(n)
O(n)
将并查集操作复杂度视作
O
(
1
)
O(1)
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];
}
}