树可以看成是一个连通且 无环 的 无向 图。
给定往一棵 n 个节点 (节点值 1~n) 的树中添加一条边后的图。添加的边的两个顶点包含在 1 到 n 中间,且这条附加的边不属于树中已存在的边。图的信息记录于长度为 n 的二维数组 edges ,edges[i] = [ai, bi] 表示图中在 ai 和 bi 之间存在一条边。
请找出一条可以删去的边,删除后可使得剩余部分是一个有着 n 个节点的树。如果有多个答案,则返回数组 edges 中最后出现的边
使用并查集即可,官方给的解答已经很棒了.
在一棵树中,边的数量比节点的数量少 11。如果一棵树有 nn 个节点,则这棵树有 n-1n−1 条边。这道题中的图在树的基础上多了一条附加的边,因此边的数量也是 nn。
树是一个连通且无环的无向图,在树中多了一条附加的边之后就会出现环,因此附加的边即为导致环出现的边。
可以通过并查集寻找附加的边。初始时,每个节点都属于不同的连通分量。遍历每一条边,判断这条边连接的两个顶点是否属于相同的连通分量。
- 如果两个顶点属于不同的连通分量,则说明在遍历到当前的边之前,这两个顶点之间不连通,因此当前的边不会导致环出现,合并这两个顶点的连通分量。
- 如果两个顶点属于相同的连通分量,则说明在遍历到当前的边之前,这两个顶点之间已经连通,因此当前的边导致环出现,为附加的边,将当前的边作为答案返回。
public int[] findRedundantConnection(int[][] edges) {
int n = edges.length;
int[] parent = new int[n+1];
for (int i=1;i<=n;i++) parent[i]=i;
for (int i=0;i<n;i++){
int[] edge = edges[i];
int node1 = edge[0],node2 = edge[1];
if(find(parent,node1)!=find(parent,node2)){
union(parent,node1,node2);
}else{
return edge;
}
}
return new int[0];
}
public void union(int[] parent,int node1,int node2){
//将node1的代表结点赋给node2
parent[find(parent,node2)] = find(parent,node1);
}
public int find(int[] parent,int index){
while(parent[index]!=index){
index = parent[index];
}
return parent[index];
}