leetcode Clone Graph

  这周我们学习了关于图的相关算法,于是我在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容器的代码。

这里写图片描述

/**
 * Definition for undirected graph.
 * struct UndirectedGraphNode {
 *     int label;
 *     vector<UndirectedGraphNode *> neighbors;
 *     UndirectedGraphNode(int x) : label(x) {};
 * };
 */
#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;
        }
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值