题目
给你无向 连通 图中一个节点的引用,请你返回该图的 深拷贝(克隆)。
图中的每个节点都包含它的值 val(int) 和其邻居的列表(list[Node])。
class Node {
public int val;
public List<Node> neighbors;
}
测试用例格式:
简单起见,每个节点的值都和它的索引相同。例如,第一个节点值为 1(val = 1),第二个节点值为 2(val = 2),以此类推。该图在测试用例中使用邻接列表表示。
邻接列表 是用于表示有限图的无序列表的集合。每个列表都描述了图中节点的邻居集。
给定节点将始终是图中的第一个节点(值为 1)。你必须将 给定节点的拷贝 作为对克隆图的引用返回。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/clone-graph
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析
克隆图,仍然以DFS为解法。遍历每一个节点,如果没访问过,就生成新的节点,然后遍历其Neighbors,递归访问,如果访问过,则直接返回存储的这个节点。
解法
/**
* Definition for a Node.
* type Node struct {
* Val int
* Neighbors []*Node
* }
*/
func cloneGraph(node *Node) *Node {
var visited = make(map[*Node]*Node)
return dfs(visited, node)
}
func dfs(visited map[*Node]*Node, cur *Node) *Node {
if cur == nil {
return nil
}
if n, ok:= visited[cur]; ok {
return n
}
// 生成新的节点
clone := &Node{Val: cur.Val, Neighbors: make([]*Node, len(cur.Neighbors))}
// 存储已经访问过了
visited[cur] = clone
// 遍历其neighbors,然后递归调用dfs。
for i, n := range cur.Neighbors {
clone.Neighbors[i] = dfs(visited, n)
}
return clone
}