复制一张图,图的顶点的数据结构如下:
struct UndirectedGraphNode {
int label;
vector<UndirectedGraphNode *> neighbors;
UndirectedGraphNode(int x) : label(x) {};
};
每个顶点存储了顶点的数值和与此顶点邻接的顶点集合。其中,每个顶点的label各不相同。
这道题可以采用图的广度优先遍历来解决。对图遍历两次,第一次复制顶点的label值,第二次复制顶点之间的邻接关系。由于每个顶点的label各不相同,可以采用map作为哈希存储结构,键值即顶点的label。代码如下:
UndirectedGraphNode *cloneGraph(UndirectedGraphNode *node) {
if( node == NULL )
return NULL;
queue<UndirectedGraphNode*> q;
q.push(node);
map<int,UndirectedGraphNode*> nodeMap;
nodeMap[node->label] = new UndirectedGraphNode(node->label);
while( !q.empty() ){
UndirectedGraphNode* cur = q.front();
q.pop();
for(int i=0;i < cur->neighbors.size(); i++){
if( nodeMap.find(cur->neighbors[i]->label) == nodeMap.end() ){
nodeMap[cur->neighbors[i]->label] = new UndirectedGraphNode(cur->neighbors[i]->label);
q.push(cur->neighbors[i]);
}
}
}
q.push(node);
for(int i=0;i < node->neighbors.size(); i++){
nodeMap[node->label]->neighbors.push_back(nodeMap[node->neighbors[i]->label]);
}
while( !q.empty() ){
UndirectedGraphNode* cur = q.front();
q.pop();
for(int i=0;i < cur->neighbors.size(); i++){
if( nodeMap[cur->neighbors[i]->label]->neighbors.size()==0 && cur->neighbors[i]->neighbors.size() > 0 ){
for(int j=0;j < cur->neighbors[i]->neighbors.size(); j++){
nodeMap[cur->neighbors[i]->label]->neighbors.push_back(nodeMap[cur->neighbors[i]->neighbors[j]->label]);
}
q.push(cur->neighbors[i]);
}
}
}
return nodeMap[node->label];
}
第一次遍历的时候,通过在map中查找节点的label来判断节点是否已访问:如果节点未访问,则在map中复制节点,并将节点入队列。第二次遍历的时候,通过判断map中与当前访问节点label相同的节点是否已被赋予邻接关系来判断节点是否已访问:如果没有,则复制邻接关系,并将节点入队列。