leetcode —— 133. 克隆图

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

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

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

解题思路:使用广度优先遍历,首先将初始结点保存在一个队列中,然后在字典中新建一个初始结点的复制节点copy_node,其中copy_node.val = node.val,但是copy_node.[]为空,其具体邻居需要在队列遍历的时候放进去。

使用字典保存已经遇到的节点的复制节点。具体Python代码如下:

"""
# 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 node
        queue = collections.deque([node])  # 初始节点放进队列中
        dicts = dict()
        dicts[node] = Node(node.val,[])  # 对节点进行复制,但是节点的邻居为空
        while queue:
            num = queue.popleft()
            for neighbor in num.neighbors:  # 对于原始数据的邻居
                if neighbor not in dicts.keys():  # 如果没有在字典中出现
                    dicts[neighbor] = Node(neighbor.val,[])    # 复制节点到字典中
                    queue.append(neighbor)  # 将原节点放进队列中进行遍历
                dicts[num].neighbors.append(dicts[neighbor])  # 对复制节点的邻居进行更新
        return dicts[node]

你可能感兴趣的:(LeetCode算法题)