leetcode 探索隊列與棧 克隆圖

題目

給你無向 連通 圖中一個節點的引用,請你返回該圖的 深拷貝(克隆)。
圖中的每個節點都包含它的值 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
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章