clone_graph.py

# -*- coding: utf-8 -*-

class Solution:
 # @param node, a undirected graph node
 # @return a undirected graph node
 def __init__(self):
 self.dict = {} # 存放已复制的节点
 
 def cloneGraph(self, node):
 if not node:
 return None
 # write your code here
 if node in self.dict:
 return self.dict[node] # 该节点已复制
 ret = UndirectedGraphNode(node.label)
 self.dict[node] = ret
 for neighbor in node.neighbors:
 ret.neighbors.append(self.cloneGraph(neighbor))
 return ret