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
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章