moonlightshadow123
6/19/2017 - 9:35 AM

133. Clone Graph

  1. Clone Graph
# Definition for a undirected graph node
# class UndirectedGraphNode:
#     def __init__(self, x):
#         self.label = x
#         self.neighbors = []

class Solution:
    # @param node, a undirected graph node
    # @return a undirected graph node
    def cloneGraph(self, node):
        if node == None:
            return None
        queue = []
        self.dict = {}
        cloneNode = UndirectedGraphNode(node.label)
        queue.append(node)
        self.dict[node] = cloneNode
        while len(queue) != 0:
            curNode = queue[0]
            del queue[0]
            curCloneNode = self.getOrCreate(curNode)
            for eachNode in curNode.neighbors:
                if self.dict.has_key(eachNode):
                    curCloneNode.neighbors.append(self.dict[eachNode])
                else:
                    curCloneNode.neighbors.append(self.getOrCreate(eachNode))
                    queue.append(eachNode)
        return cloneNode
    def getOrCreate(self, node):
        if self.dict.has_key(node):
            return self.dict[node]
        else:
            cloneNode = UndirectedGraphNode(node.label)
            self.dict[node] = cloneNode
            return cloneNode

https://leetcode.com/problems/clone-graph/#/description

Clone an undirected graph. Each node in the graph contains a label and a list of its neighbors.

OJ's undirected graph serialization: Nodes are labeled uniquely.

We use # as a separator for each node, and , as a separator for node label and each neighbor of the node. As an example, consider the serialized graph {0,1,2#1,2#2,2}.

The graph has a total of three nodes, and therefore contains three parts as separated by #.

  1. First node is labeled as 0. Connect node 0 to both nodes 1 and 2.
  2. Second node is labeled as 1. Connect node 1 to node 2.
  3. Third node is labeled as 2. Connect node 2 to node 2 (itself), thus forming a self-cycle.

Visually, the graph looks like the following:

       1
      / \
     /   \
    0 --- 2
         / \
         \_/