复杂数据结构(并查集)算法简略记录
1、并查集
并查集(union-find或disjoint set)可以动态联通两个点,并且可以快速判断两个点是否联通,求最小生成树;由一个数组、两个函数(查找一个数的根、合并路线)组成,常用来解决一些不相交集合的合并与查找问题
- 合并(Union):把两个不相交的集合合并为一个集合。
- 查询(Find):查询两个元素是否在同一个集合中。
解题思路
- 如果两个顶点属于不同的连通分量,则说明在遍历到当前的边之前,这两个顶点之间不连通,因此当前的边不会导致环出现,合并这两个顶点的连通分量。
- 如果两个顶点属于相同的连通分量,则说明在遍历到当前的边之前,这两个顶点之间已经连通,因此当前的边导致环出现,为附加的边,将当前的边作为答案返回。
Java解答
class Solution {
public int[] findRedundantConnection(int[][] edges) {
int nodesCount = edges.length;
int[] parent = new int[nodesCount+1];
for (int i = 1; i <= nodesCount; i++) {
parent[i] = i;
}
for (int i = 0; i < nodesCount; i++) {
int[] edge = edges[i];
int node1 = edge[0], node2 = edge[1];
if (find(parent, node1