[LintCode] Clone Graph

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

How we serialize an undirected graph:

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:

 

Algorithm:

If we copy graph nodes as we do a bfs, there will be problems if the graph has self cycle. Also since the input graph is undirected, for each edge, we need to add it to both nodes.The regular bfs skips nodes that have already been visited, which makes adding an edge to both nodes' neighbors list difficult.

 

A simpler solution is as follows. O(n + m) runtime, O(n) space

1. bfs to get all original nodes.

2. copy all nodes and create a mapping between the original nodes and copied nodes.

3. Iterate the original nodes list from step 1 and construct the edges for the copied graph using the mapping relation from step 2.

/**
 * Definition for undirected graph.
 * class UndirectedGraphNode {
 *     int label;
 *     ArrayList<UndirectedGraphNode> neighbors;
 *     UndirectedGraphNode(int x) { label = x; neighbors = new ArrayList<UndirectedGraphNode>(); }
 * };
 */
public class Solution {
    /**
     * @param node: A undirected graph node
     * @return: A undirected graph node
     */
    public UndirectedGraphNode cloneGraph(UndirectedGraphNode node) {
        if(node == null)
        {
            return null;
        }
        
        //use bfs algorithm to traverse the graph and get all nodes 
        ArrayList<UndirectedGraphNode> nodes = getNodes(node);
        HashMap<UndirectedGraphNode, UndirectedGraphNode> copyMapping = new HashMap<UndirectedGraphNode, UndirectedGraphNode>();
        //copy nodes
        for(UndirectedGraphNode n : nodes)
        {
            copyMapping.put(n, new UndirectedGraphNode(n.label));
        }
        
        //copy neighbors(edges)
        for(UndirectedGraphNode n : nodes)
        {
            UndirectedGraphNode newNode = copyMapping.get(n);
            for(UndirectedGraphNode neighbor : n.neighbors)
            {
                UndirectedGraphNode newNeighbor = copyMapping.get(neighbor);
                newNode.neighbors.add(newNeighbor);
            }
        }
        return copyMapping.get(node);
    }
    private ArrayList<UndirectedGraphNode> getNodes(UndirectedGraphNode node)
    {
        Queue<UndirectedGraphNode> queue = new LinkedList<UndirectedGraphNode>();
        HashSet<UndirectedGraphNode> visited = new HashSet<UndirectedGraphNode>();
        
        queue.offer(node);
        visited.add(node);
        while(!queue.isEmpty())
        {
            UndirectedGraphNode head = queue.poll();
            for(UndirectedGraphNode neighbor : head.neighbors)
            {
                if(!visited.contains(neighbor))
                {
                    visited.add(neighbor);
                    queue.offer(neighbor);
                }
            }
        }
        return new ArrayList<UndirectedGraphNode>(visited);
    }
}

 

Related Problems

Six Degrees

Clone Binary Tree

Route Between Two Nodes in Graph