LeetCode 133. 克隆图

Posted 2021-10-12 Blocking The Sky

给你无向 连通 图中一个节点的引用，请你返回该图的 深拷贝（克隆）。

图中的每个节点都包含它的值 val（int） 和其邻居的列表（list[Node]）。

深度优先搜索

class Solution {
public:
    unordered_map<Node*,Node*> visited;  //记录原节点对应的新图中的克隆节点
    Node* cloneGraph(Node* node)   //递归一定要弄清楚函数的返回值是什么，这儿返回值是node节点对应的克隆图中的节点
    {
        if(!node) 
            return NULL;  //空节点直接返回
        
        if(visited.find(node)!=visited.end())  //如果已经克隆好了，那么我们直接用就行，不用再克隆了
            return visited[node];
        
        Node* cloneNode=new Node(node->val);   //克隆新节点，注意要实现深拷贝，不能直接传原节点的邻居集
        visited[node]=cloneNode;

        for(int i=0;i<node->neighbors.size();i++)     //对于node的邻居集中的每个节点，都进行克隆，然后把克隆的节点加入到cloneNode的邻居集中
            cloneNode->neighbors.push_back(cloneGraph(node->neighbors[i]));  //cloneGraph返回的是对应的克隆节点

        return cloneNode;
                                   
    }
};

宽度优先搜索

class Solution {
public:
    Node* cloneGraph(Node* node) {
        if (node == nullptr) {
            return node;
        }
        unordered_map<Node*, Node*> visited;
        // 将题目给定的节点添加到队列
        queue<Node*> Q;
        Q.push(node);
        // 克隆第一个节点并存储到哈希表中
        visited[node] = new Node(node->val);

        // 广度优先搜索
        while (!Q.empty()) {
            // 取出队列的头节点
            auto n = Q.front();
            Q.pop();
            // 遍历该节点的邻居
            for (auto& neighbor: n->neighbors) {
                if (visited.find(neighbor) == visited.end()) {
                    // 如果没有被访问过，就克隆并存储在哈希表中
                    visited[neighbor] = new Node(neighbor->val);
                    // 将邻居节点加入队列中
                    Q.push(neighbor);
                }
                // 更新当前节点的邻居列表
                visited[n]->neighbors.emplace_back(visited[neighbor]);
            }
        }

        return visited[node];
    }
};