克隆图

题目描述

给定无向连通图中一个节点的引用，返回该图的深拷贝（克隆）。图中的每个节点都包含它的值 val（Int） 和其邻居的列表（list[Node]）。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/clone-graph
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

深拷贝即复制原图中的结点和结点之间的边。用一个map记录原结点和新结点的键值对，以便于克隆图复制原图结点之间的边。广度优先搜索图，对于当前结点，复制其相邻结点（如果还没复制），用map记录和原结点的对应关系，然后再复制和当前结点的边。

代码（c++）

/*
// Definition for a Node.
class Node {
public:
    int val;
    vector<Node*> neighbors;

    Node() {}

    Node(int _val, vector<Node*> _neighbors) {
        val = _val;
        neighbors = _neighbors;
    }
};
*/
class Solution {
public:
    Node* cloneGraph(Node* node) {
        if(node==NULL) return NULL;
        map<Node*,Node*> mp;
        queue<Node*> old_q;
        old_q.push(node);
        vector<Node*> neighbors;
        Node* res=new Node(node->val,neighbors);
        mp[node]=res;
        while(!old_q.empty()){
            Node* old_temp=old_q.front();
            old_q.pop();
            for(int i=0;i<old_temp->neighbors.size();i++){
                if(mp.find(old_temp->neighbors[i])==mp.end()){
                    vector<Node*> neighbors;
                    Node* new_node=new Node(old_temp->neighbors[i]->val,neighbors);
                    mp[old_temp->neighbors[i]]=new_node;
                    old_q.push(old_temp->neighbors[i]);
                }
                mp[old_temp]->neighbors.push_back(mp[old_temp->neighbors[i]]);
            }
        }
        return res;
    }
};