137-克隆图
克隆一张无向图,图中的每个节点包含一个 label 和一个列表 neighbors。
数据中如何表示一个无向图?http://www.lintcode.com/help/graph/
你的程序需要返回一个经过深度拷贝的新图。这个新图和原图具有同样的结构,并且对新图的任何改动不会对原图造成任何影响。样例
比如,序列化图 {0,1,2#1,2#2,2} 共有三个节点, 因此包含两个个分隔符#。
- 第一个节点label为0,存在边从节点0链接到节点1和节点2
- 第二个节点label为1,存在边从节点1连接到节点2
- 第三个节点label为2,存在边从节点2连接到节点2(本身),从而形成自环。
我们能看到如下的图:
标签
宽度优先搜索 脸书
思路
使用宽度优先搜索遍历原图的同时创建新图。使用 oldToNew 保存原图与新图同个节点的对应关系,用 isVisited 保存原图的某点是否被遍历。具体过程如下:
- 对原图的第一个节点 node,新建一个节点 newNode 与 node 对应,即oldToNew[node] = newNode,并标记 node 已遍历,即 isVisited[node] = true,将 node 入队
- 队列不为空时:
队首元素 node 出队,遍历与 node 相连的节点,即 node->neighbors[i],
若 node->neighbors[i] 未被遍历过,建立新节点与其对应,标记此结点以便利,将新节点连入新图
若 node->neighbors[i] 已被遍历,将此结点对应的新节点连入新图
code
/**
* Definition for undirected graph.
* struct UndirectedGraphNode {
* int label;
* vector<UndirectedGraphNode *> neighbors;
* UndirectedGraphNode(int x) : label(x) {};
* };
*/
class Solution {
public:
/**
* @param node: A undirected graph node
* @return: A undirected graph node
*/
UndirectedGraphNode *cloneGraph(UndirectedGraphNode *node) {
// write your code here
if(node == NULL) {
return node;
}
queue<UndirectedGraphNode*> willVisit;
map<UndirectedGraphNode*, bool> isVisited;
map<UndirectedGraphNode*, UndirectedGraphNode*> oldToNew;
UndirectedGraphNode * newNode = new UndirectedGraphNode(node->label);
isVisited[node] = true;
oldToNew[node] = newNode;
willVisit.push(node);
while(!willVisit.empty()) {
node = willVisit.front();
willVisit.pop();
for(int i=0; i<node->neighbors.size(); i++) {
if(isVisited[node->neighbors[i]] == false) {
UndirectedGraphNode * temp = new UndirectedGraphNode(node->neighbors[i]->label);
isVisited[node->neighbors[i]] = true;
oldToNew[node->neighbors[i]] = temp;
oldToNew[node]->neighbors.push_back(temp);
willVisit.push(node->neighbors[i]);
}
else {
oldToNew[node]->neighbors.push_back(oldToNew[node->neighbors[i]]);
}
}
}
return newNode;
}
};