您的位置:首页 > 其它

133. Clone Graph

2016-04-20 00:48 274 查看
Clone an undirected graph. Each node in the graph contains a 
label
 and
a list of its 
neighbors
.

Solution 1 BFS

//BFS
public UndirectedGraphNode cloneGraph(UndirectedGraphNode node) {
if(node == null){
return null;
}
Queue<UndirectedGraphNode> q = new LinkedList<UndirectedGraphNode>();
Map<UndirectedGraphNode, UndirectedGraphNode> map = new HashMap<UndirectedGraphNode, UndirectedGraphNode>();
map.put(node,new UndirectedGraphNode(node.label));
q.add(node);
while(!q.isEmpty()){
UndirectedGraphNode temp = q.poll();
for(UndirectedGraphNode neighbour : temp.neighbors){
if(!map.containsKey(neighbour)){
map.put(neighbour, new UndirectedGraphNode(neighbour.label));
q.add(neighbour);
}
map.get(temp).neighbors.add(map.get(neighbour));
}
}
return map.get(node);
}

Solution 2 DFS Recursive
//DFS
Map<UndirectedGraphNode, UndirectedGraphNode> map = new HashMap<UndirectedGraphNode, UndirectedGraphNode>();
public UndirectedGraphNode cloneGraph2(UndirectedGraphNode node) {
if(node == null){
return null;
}
if(map.containsKey(node)){
return map.get(node);
}
UndirectedGraphNode temp = new UndirectedGraphNode(node.label);
map.put(node, temp);
for(UndirectedGraphNode n : node.neighbors){
temp.neighbors.add(cloneGraph2(n));
}
return temp;
}

Solution 3 DFS Iterative
//Iterative DFS
public UndirectedGraphNode cloneGraph3(UndirectedGraphNode root) {
if (root == null) return null;

// use a stack to help DFS
Stack<UndirectedGraphNode> stack = new Stack<UndirectedGraphNode>();
stack.push(root);

// use a map to save cloned nodes
Map<UndirectedGraphNode, UndirectedGraphNode> map = new HashMap<UndirectedGraphNode, UndirectedGraphNode>();

// clone the root
map.put(root, new UndirectedGraphNode(root.label));

while (!stack.isEmpty()) {
UndirectedGraphNode node = stack.pop();

// handle the neighbors
for (UndirectedGraphNode neighbor : node.neighbors) {
if (!map.containsKey(neighbor)) {
// clone the neighbor
map.put(neighbor, new UndirectedGraphNode(neighbor.label));
// add it to the next level
stack.push(neighbor);
}

// copy the neighbor
map.get(node).neighbors.add(map.get(neighbor));
}
}

return map.get(root);
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: