題目
給你無向 連通 圖中一個節點的引用,請你返回該圖的 深拷貝(克隆)。
圖中的每個節點都包含它的值 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
}