题目描述
方法思路
class Solution {
//Runtime: 2 ms, faster than 100.00%
//Memory Usage: 41.2 MB, less than 11.55%
public int[] findRedundantConnection(int[][] edges) {
int[] parent = new int[2001];
//初始化各个孤立节点的根节点为其本身
for (int i = 0; i < parent.length; i++) parent[i] = i;
for (int[] edge: edges){
int f = edge[0], t = edge[1];
//如果两个节点的根节点相同,则返回这条边
if (find(parent, f) == find(parent, t)) return edge;
//若根节点不同,则修改第一个节点的根节点,联通两个子图,此处制定了节点依附规则。
//保证了如果有多个边是冗余边,则返回最后出现在二维数组中的边。
else parent[find(parent, f)] = find(parent, t);
}
return new int[2];
}
//查找根节点并返回
private int find(int[] parent, int f) {
if (f != parent[f]) {
parent[f] = find(parent, parent[f]);
}
return parent[f];
}
}