Leetcode练习：克隆图

题目：

给定无向连通图中一个节点的引用，返回该图的深拷贝（克隆）。图中的每个节点都包含它的值 val（Int） 和其邻居的列表（list[Node]）。

示例：

 

输入：
{"$id":"1","neighbors":[{"$id":"2","neighbors":[{"$ref":"1"},{"$id":"3","neighbors":[{"$ref":"2"},{"$id":"4","neighbors":[{"$ref":"3"},{"$ref":"1"}],"val":4}],"val":3}],"val":2},{"$ref":"4"}],"val":1}

解释：
节点 1 的值是 1，它有两个邻居：节点 2 和 4 。
节点 2 的值是 2，它有两个邻居：节点 1 和 3 。
节点 3 的值是 3，它有两个邻居：节点 2 和 4 。
节点 4 的值是 4，它有两个邻居：节点 1 和 3 。

 

提示：

    节点数介于 1 到 100 之间。
    无向图是一个简单图，这意味着图中没有重复的边，也没有自环。
    由于图是无向的，如果节点 p 是节点 q 的邻居，那么节点 q 也必须是节点 p 的邻居。
    必须将给定节点的拷贝作为对克隆图的引用返回。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/clone-graph
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

思路：深度优先遍历该图结构直到遍历完成。遍历过程中，将所有边结构记录下来，同时每遇到一个新节点就建立一个新节点，新节点neighbors列表为空。之后通过列表记录重建节点之间的连接关系。

 

代码：

"""
# Definition for a Node.
class Node(object):
    def __init__(self, val, neighbors):
        self.val = val
        self.neighbors = neighbors
"""
class Solution(object):
    def cloneGraph(self, node):
        """
        :type node: Node
        :rtype: Node
        """
        # traverse to get the node values and edges
        stack = [node]
        edges = []
        nodes = {}
        while(stack):
            current_node = stack[-1]
            if current_node.val not in nodes:
                nodes[current_node.val] = Node(current_node.val, [])
            if not current_node.neighbors:
                stack.remove(current_node)
                continue
            next_node = current_node.neighbors[0]
            edges.append((current_node.val, next_node.val))
            current_node.neighbors.remove(next_node)
            #next_node.neighbors.remove(current_node)
            stack.append(next_node)
        
        for key in nodes.keys():
            for edge in edges:
                val1 = edge[0]
                val2 = edge[1]
                if val1 == key:
                    nodes[val1].neighbors.append(nodes[val2])
            
        return nodes[node.val]