Clone Graph [leetcode] dfs和bfs

变量unordered_map<UndirectedGraphNode *, UndirectedGraphNode *> cloneMap;

因为会有环,所以需要cloneMap记录旧的节点和新的节点对。

还需要一个visited记录已经访问过的节点,可以和cloneMap合并在一起。


DFS

在克隆某个Node时,首先放入map中。

之后遍历neighbors。如果neighbors不在map里,即没有访问过,则递归调用cloneGraph函数。

最后返回克隆过的节点。

    UndirectedGraphNode *cloneGraph(UndirectedGraphNode *node) {
        if (node == NULL) return NULL;
        if (cloneMap.find(node) != cloneMap.end()) 
            return cloneMap[node];
        cloneMap[node] = new UndirectedGraphNode(node->label);
        cloneMap[node]->neighbors = vector<UndirectedGraphNode *>();
        for (int i = 0; i < node->neighbors.size(); i++)
        {
            UndirectedGraphNode * curNode = node->neighbors[i];
            UndirectedGraphNode * newNode = cloneGraph(curNode);
            cloneMap[node]->neighbors.push_back(newNode);
        }
        return cloneMap[node];
    }

BFS

使用一个queue记录要访问的节点。

因为不能在访问queue中节点的时候再创建对应的node==>因为在处理父亲节点的时候需同时创建好其neighbors

所以需要在将不在map中的neighbors(没有访问过的子节点)放入queue的同时创建克隆的节点,并且放入map里。

    UndirectedGraphNode *cloneGraph(UndirectedGraphNode *node) {
        if (node == NULL) return NULL;
        vector<UndirectedGraphNode *> queue;
        queue.push_back(node);
        cloneMap[node] = new UndirectedGraphNode(node->label);
        cloneMap[node]->neighbors = vector<UndirectedGraphNode *>();
        
        for (int i = 0; i < queue.size(); i++)
        {
            UndirectedGraphNode * newNode = cloneMap[queue[i]];
            for (int j = 0; j < queue[i]->neighbors.size(); j++)
            {
                UndirectedGraphNode * curNb = queue[i]->neighbors[j];
                if (cloneMap.find(curNb) == cloneMap.end())
                {
                    queue.push_back(curNb);
                    UndirectedGraphNode * newNb = new UndirectedGraphNode(curNb->label);
                    newNb->neighbors = vector<UndirectedGraphNode *>();
                    cloneMap[curNb] = newNb;
                }
                newNode->neighbors.push_back(cloneMap[curNb]);
            }
        }
        return cloneMap[node];
    }


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值