这周我们学习了关于图的相关算法,于是我在leetcode也找了一道关于图的题。这题算是我在leetcode遇到的题中最简单的了(话说回来,现在做过的题还是极少的)。这题我觉得题干有点啰嗦,大部分都是和解题无关的话语。简单地说,它就是要让你构建一个和原先相同的图。给出的结构体说明每个节点都有一个int类型的label,还有一个vector容器存储与它直接相连的节点。我们要做的就是,根据输入的一个原图的节点,遍历这个图,把访问的每个节点复制一遍就行了,个人觉得这题本质上只是考察怎样遍历一个图而已……
为了解决这个问题,我最初的想法是对原图进行BFS,用一个vector容器存放复制出来的节点们的指针,在访问原图第一个节点的时候,复制这个节点,并且存储在vector容器里。访问一个节点的相邻节点时,判断相邻节点有没有在vector容器内,不在的话就复制然后存进容器,并且添加到这个节点的复制节点(vector容器内)的neighbors里,而且在队列中添加相邻节点。在的话就不存进容器,也不需要添加进队列,但还是要加进neighbors里。所以vector容器的作用除了保存复制的各个节点之外,还相当于BFS算法中的visited数组,可以判断节点是不是第一次访问,需不需要加进队列。最后我用这个方法过了这道题,但是运行时间只处于所有提交的百分之零点几,我估计是每次都在vector中查找影响了效率。于是我采用了第二种方法。
这次我将vector换成了map,之前我根据vector中每个节点对应的label值进行节点的查找,现在换成了map,我就可以把label当成它的键值,这样就可以调用map的find函数,不需要自己定义了。提交上去之后,运行时间打败了80%左右的提交,我觉得是map红黑树的存储结构所起的作用,以下是我用map容器的代码。
![这里写图片描述](https://img-blog.csdn.net/20170923162818566?watermark/2/text/aHR0cDovL2Jsb2cuY3Nkbi5uZXQvcXFfMzg3Njc2Njg=/font/5a6L5L2T/fontsize/400/fill/I0JBQkFCMA==/dissolve/70/gravity/SouthEast)
#include <queue>
#include <map>
using namespace std;
class Solution {
public:
UndirectedGraphNode *cloneGraph(UndirectedGraphNode *node) {
if (node == NULL) {
return NULL;
} else {
queue<UndirectedGraphNode*> q;
map<int, UndirectedGraphNode*> m;
map<int, UndirectedGraphNode*>::iterator iter;
UndirectedGraphNode* top = new UndirectedGraphNode(node->label);
m.insert(pair<int, UndirectedGraphNode*> (node->label, top));
q.push(node);
while (!q.empty()) {
UndirectedGraphNode* temp = q.front();
UndirectedGraphNode* copyNode = m.find(temp->label)->second;
q.pop();
for (int i = 0; i < temp->neighbors.size(); i++) {
int labelNum = temp->neighbors[i]->label;
iter = m.find(labelNum);
if (iter == m.end()) {
UndirectedGraphNode* copyNeigh = new UndirectedGraphNode(labelNum);
copyNode->neighbors.push_back(copyNeigh);
m.insert(pair<int, UndirectedGraphNode*> (labelNum, copyNeigh));
q.push(temp->neighbors[i]);
} else {
copyNode->neighbors.push_back(iter->second);
}
}
}
return m.begin()->second;
}
}
};