题目大意:生成无向简单图的克隆图(图结构相同,结点是新建的)
分析:DFS+MAP。
DFS的趴很好思考,就是克隆当前结点,然后遍历它的邻接表dfs克隆即可。
但是涉及一个问题——需要防止重复克隆同一个结点。所以需要类似visit数组一样的数据结构进行标记。
采用map<key,value>=map<node,newNode>,记录node的克隆结点。访问node时先判断是否克隆过,如果克隆过就直接返回,否则才新建newNode并加入map。
代码:
/*
// Definition for a Node.
class Node {
public:
int val;
vector<Node*> neighbors;
Node() {
val = 0;
neighbors = vector<Node*>();
}
Node(int _val) {
val = _val;
neighbors = vector<Node*>();
}
Node(int _val, vector<Node*> _neighbors) {
val = _val;
neighbors = _neighbors;
}
};
*/
class Solution {
public:
map<Node*,Node*> cloneNode;
Node* cloneGraph(Node* node) {
if(node == NULL) return NULL;
if(cloneNode.count(node)) return cloneNode[node];
Node* newNode = new Node(node->val);
cloneNode[node] = newNode;
for(int i = 0;i < node->neighbors.size();i++){
newNode->neighbors.push_back(cloneGraph(node->neighbors[i]));
}
return newNode;
}
};