Leetcode 133.克隆图(Clone Graph)

Leetcode 133.克隆图

1 题目描述(Leetcode题目链接)

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

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

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

Leetcode 133.克隆图(Clone Graph)_第1张图片

测试用例格式:

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

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

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

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

2 题解

  深度优先搜索,为了处理无向图,需要使用个字典记录一下拷贝出来的对应节点。

"""
# Definition for a Node.
class Node:
    def __init__(self, val = 0, neighbors = []):
        self.val = val
        self.neighbors = neighbors
"""
class Solution:
    def cloneGraph(self, node: 'Node') -> 'Node':
        if not node:
            return None
        seen = {}		# 记录原节点和拷贝节点的对应关系
        
        def dfs(root, node):
            for child in node.neighbors:
                if child in seen:
                    root.neighbors.append(seen[child])
                    continue
                new = Node(child.val)
                seen[child] = new
                root.neighbors.append(new)
                dfs(new, child)

        root = Node(node.val)
        seen[node] = root
        dfs(root, node)
        return root

你可能感兴趣的:(Leetcode)