解题思路:
这道题目,一开始看了几遍都没看懂题意,后来找网上的答案才明白是要干啥。其实就是做一次深拷贝的实现。也是比较典型的深度优先搜索问题。
使用C++中的unordered_map<Node*, Node*> 来克隆节点,在访问Node时,先判断是否已经对该节点做过克隆,如果有,则直接返回该克隆节点即可,如果没有,则执行程序,克隆该节点:包括val值和neighbors,在克隆其neighbors时,可能会遇到未克隆的节点,所以需要递归调用克隆函数。
/*
// 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) {
unordered_map<Node*, Node*> mp;
return dfs(node, mp);
}
Node* dfs(Node* node, unordered_map<Node*, Node*>& mp){
if(!node) return NULL;
if(mp.count(node)) return mp[node];
Node* tmp = new Node(node->val);
mp[node] = tmp;
for(Node* neighbor: node->neighbors){
tmp->neighbors.push_back(dfs(neighbor, mp));
}
return tmp;
}
};