克隆图

81 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 2 月更文挑战」的第 8 天,点击查看活动详情

给你无向 连通 图中一个节点的引用,请你返回该图的 深拷贝(克隆)。

图中的每个节点都包含它的值 val(int) 和其邻居的列表(list[Node])。

class Node { public int val; public List neighbors; }

测试用例格式:

简单起见,每个节点的值都和它的索引相同。例如,第一个节点值为 1(val = 1),第二个节点值为 2(val = 2),以此类推。该图在测试用例中使用邻接列表表示。

邻接列表 是用于表示有限图的无序列表的集合。每个列表都描述了图中节点的邻居集。

给定节点将始终是图中的第一个节点(值为 1)。你必须将 给定节点的拷贝 作为对克隆图的引用返回。

示例 1:

输入:adjList = [[2,4],[1,3],[2,4],[1,3]] 输出:[[2,4],[1,3],[2,4],[1,3]] 解释: 图中有 4 个节点。 节点 1 的值是 1,它有两个邻居:节点 2 和 4 。 节点 2 的值是 2,它有两个邻居:节点 1 和 3 。 节点 3 的值是 3,它有两个邻居:节点 2 和 4 。 节点 4 的值是 4,它有两个邻居:节点 1 和 3 。

示例 2:

输入:adjList = [[]] 输出:[[]] 解释:输入包含一个空列表。该图仅仅只有一个值为 1 的节点,它没有任何邻居。

示例 3:

输入:adjList = [] 输出:[] 解释:这个图是空的,它不含任何节点。

示例 4:

输入:adjList = [[2],[1]] 输出:[[2],[1]]

提示:

节点数不超过 100 。
每个节点值 Node.val 都是唯一的,1 <= Node.val <= 100。
无向图是一个简单图,这意味着图中没有重复的边,也没有自环。
由于图是无向的,如果节点 p 是节点 q 的邻居,那么节点 q 也必须是节点 p 的邻居。
图是连通图,你可以从给定节点访问到所有节点。

思路

  题意是对一个无向图进行深度拷贝,可以考虑先拷贝节点,再新旧节点之间进行映射,使我们能够通过旧节点找到新节点,再拷贝边,即将旧节点的邻居所映射的新节点,作为新邻居赋给该旧节点所映射的新节点。

需要注意的是,很容易将旧邻居直接赋给新节点,这样是错误的,因为旧节点对应的邻居也是旧的,需要各自找到映射的新邻居。

代码

   const cloneGraph = function(node) {
   const helper = (node) => {
       if (!node) {
           return null
       }
       if (map.has(node)) {
           return map.get(node)
       }
       const { val, neighbors } = node
       const root = new Node(val)
       map.set(node, root)
       for (let i = 0; i < neighbors.length; i++) {
           root.neighbors[i] = helper(neighbors[i])
       }
       return root
   }
   const map = new Map()
   return helper(node)
}