LeetCode: Clone Graph

最新推荐文章于 2019-01-11 00:33:31 发布

AIvin24

最新推荐文章于 2019-01-11 00:33:31 发布

阅读量793

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/AIvin24/article/details/38981447

版权

LeetCode 专栏收录该内容

144 篇文章 0 订阅

订阅专栏

思路：刚开始用一个DFS方法，不过这题目最好用BFS方法，这篇博客介绍的思路不错，而且代码精简，套用这个模式的code如下：

class Solution {
    unordered_map<int, UndirectedGraphNode*> m;  // m 存储新图中的节点label 和 指针位置
public:
    UndirectedGraphNode *cloneGraph(UndirectedGraphNode *node) {
        if(node == NULL)
            return node;
        queue<UndirectedGraphNode *> Q;
        Q.push(node);
        while(!Q.empty()){
            UndirectedGraphNode * qFront = Q.front();
            Q.pop();
            if(m.find(qFront->label) == m.end())
                m[qFront->label] = new UndirectedGraphNode(qFront->label); 
            for(int i = 0;i < (qFront->neighbors).size();i++){
                if(m.find((qFront->neighbors)[i]->label) == m.end()){// 作为是否已经添加过的标记
                    m[(qFront->neighbors)[i]->label] = new UndirectedGraphNode((qFront->neighbors)[i]->label);
                    Q.push((qFront->neighbors)[i]);
                }
                m[qFront->label]->neighbors.push_back(m[qFront->neighbors[i]->label]);
            }
        }
        return m[node->label];
    }
};