[LeetCode] 133. Clone Graph

最新推荐文章于 2024-02-04 01:58:13 发布

hiimdaosui

最新推荐文章于 2024-02-04 01:58:13 发布

阅读量213

点赞数

分类专栏： LeetCode 文章标签： leetcode dfs bfs 图论

本文链接：https://blog.csdn.net/hiimdaosui/article/details/52449293

版权

LeetCode 专栏收录该内容

184 篇文章 0 订阅

订阅专栏

思路:
这里两个思路, 第一个是DFS, 第二个是BFS. 大同小异, 都是要用一个哈希表, 哈希表第一项是原图的节点, 第二项是原图对应的新图的该节点. 哈希的好处是一能记录当前节点是否之前已经创造过新节点了, 二是维护了第二个图各节点之间的关系.

UndirectedGraphNode* dfs(unordered_map<UndirectedGraphNode*, UndirectedGraphNode*>& hashmap, 
                            UndirectedGraphNode* node) {
    if (! node) return NULL;
    if (! hashmap.count(node)) {
        hashmap[node] = new UndirectedGraphNode(node->label);
        for (int i = 0; i < node->neighbors.size(); i++) 
            hashmap[node]->neighbors.push_back(dfs(hashmap, node->neighbors[i]));
    }
    return hashmap[node];
}

UndirectedGraphNode *cloneGraph(UndirectedGraphNode *node) {
    unordered_map<UndirectedGraphNode*, UndirectedGraphNode*> hashmap;
    return dfs(hashmap, node);
}

UndirectedGraphNode *cloneGraph(UndirectedGraphNode *node) {
    if (! node) return NULL;
    unordered_map<UndirectedGraphNode*, UndirectedGraphNode*> hashmap;
    queue<UndirectedGraphNode*> q;

    UndirectedGraphNode* new_node = new UndirectedGraphNode(node->label);
    hashmap[node] = new_node;
    q.push(node);

    while (! q.empty()) {
        UndirectedGraphNode* cur = q.front();
        q.pop();
        for (int i = 0; i < cur->neighbors.size(); i++) {
            UndirectedGraphNode* curNeighbor = cur->neighbors[i];
            if (! hashmap.count(curNeighbor)) {
                hashmap[curNeighbor] = new UndirectedGraphNode(curNeighbor->label);
                q.push(curNeighbor);
            }
            hashmap[cur]->neighbors.push_back(hashmap[curNeighbor]);
        }
    }

    return new_node;
}