Clone an undirected graph. Each node in the graph contains a label and
a list of its neighbors.
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 #.
0. Connect node 0 to
both nodes 1 and 2.1. Connect node 1 to
node 2.2. Connect node 2 to
node 2 (itself), thus forming a self-cycle.Visually, the graph looks like the following:
1
/ / 0 --- 2
/ \_/
class Solution {
public:
<span style="white-space:pre"> </span>UndirectedGraphNode *cloneGraph(UndirectedGraphNode *node) {
<span style="white-space:pre"> </span>if (NULL == node) return NULL;
<span style="white-space:pre"> </span>map<int, UndirectedGraphNode *> occured;
<span style="white-space:pre"> </span>return clone(node, occured);
<span style="white-space:pre"> </span>}
<span style="white-space:pre"> </span>UndirectedGraphNode *clone(UndirectedGraphNode *node, map<int, UndirectedGraphNode *> &occured){
<span style="white-space:pre"> </span>if (occured.count(node->label) == 0){
<span style="white-space:pre"> </span>UndirectedGraphNode *newNode = new UndirectedGraphNode(node->label);
<span style="white-space:pre"> </span>occured[node->label] = newNode;
<span style="white-space:pre"> </span>for (int i = 0; i < node->neighbors.size(); i++){
<span style="white-space:pre"> </span>newNode->neighbors.push_back(clone(node->neighbors[i], occured));
<span style="white-space:pre"> </span>}
<span style="white-space:pre"> </span>return newNode;
<span style="white-space:pre"> </span>}
<span style="white-space:pre"> </span>else{
<span style="white-space:pre"> </span>return occured[node->label];
<span style="white-space:pre"> </span>}
<span style="white-space:pre"> </span>}
};class Solution {
public:
UndirectedGraphNode *cloneGraph(UndirectedGraphNode *node) {
if (NULL == node) return NULL;
map<UndirectedGraphNode *, UndirectedGraphNode *> occured;
queue<UndirectedGraphNode *> visit;
visit.push(node);
while (!visit.empty()){
UndirectedGraphNode *cur = visit.front();
visit.pop();
if (occured.count(cur) == 0){
UndirectedGraphNode *newNode = new UndirectedGraphNode(cur->label);
occured[cur] = newNode;
}
for (int i = 0; i < cur->neighbors.size(); i++){
UndirectedGraphNode *tmp = cur->neighbors[i];
if (occured.count(tmp) == 0){
UndirectedGraphNode *newNode = new UndirectedGraphNode(tmp->label);
occured[tmp] = newNode;
visit.push(tmp);
}
occured[cur]->neighbors.push_back(occured[tmp]);
}
}
return occured[node];
}
};原文地址:http://blog.csdn.net/cr_peace/article/details/41693677