[路飞]_冗余连接

684. 冗余连接

题目

树可以看成是一个连通且 无环 的 无向 图。

给定往一棵 n 个节点 (节点值 1～n) 的树中添加一条边后的图。添加的边的两个顶点包含在 1 到 n 中间，且这条附加的边不属于树中已存在的边。图的信息记录于长度为 n 的二维数组 edges ，edges[i] = [ai, bi] 表示图中在 ai 和 bi 之间存在一条边。

请找出一条可以删去的边，删除后可使得剩余部分是一个有着 n 个节点的树。如果有多个答案，则返回数组 edges 中最后出现的边。

示例1

输入: edges = [[1,2], [1,3], [2,3]]
输出: [2,3]

题解

并查集

• 构建并查集
• 枚举数据，将连通的节点合并到集合中，使他们指向同一个祖先（这句话是核心）理解如何让他们指向同一个祖先节点，是核心
• 如果两个节点有同一个祖先，说明两个节点已经有直接或者间接连通了，返回当前节点即可

代码

var findRedundantConnection = function (edges) {
  const len = edges.length;
  const list = Array(len + 1)
    .fill(0)
    .map((v, i) => i);
  for (let i = 0; i < edges.length; i++) {
    const [j, k] = edges[i];

    if (loop(j) === loop(k)) {
      return edges[i];
    } else {
      merge(j, k);
    }
  }

  function merge(j, k) {
    const x = loop(j);
    const y = loop(k);
    list[x] = y;
  }

  function loop(n) {
    if (list[n] === n) return list[n];
    return loop(list[n]);
  }
};