一、题目描述:133. 克隆图(中等)
给定无向连通图中一个节点的引用,返回该图的深拷贝(克隆)。图中的每个节点都包含它的值 val(Int) 和其邻居的列表(list[Node])。
示例:
输入:
{"$id":"1","neighbors":[{"$id":"2","neighbors":[{"$ref":"1"},{"$id":"3","neighbors":[{"$ref":"2"},{"$id":"4","neighbors":[{"$ref":"3"},{"$ref":"1"}],"val":4}],"val":3}],"val":2},{"$ref":"4"}],"val":1}解释:
节点 1 的值是 1,它有两个邻居:节点 2 和 4 。
节点 2 的值是 2,它有两个邻居:节点 1 和 3 。
节点 3 的值是 3,它有两个邻居:节点 2 和 4 。
节点 4 的值是 4,它有两个邻居:节点 1 和 3 。
提示:
节点数介于 1 到 100 之间。
无向图是一个简单图,这意味着图中没有重复的边,也没有自环。
由于图是无向的,如果节点 p 是节点 q 的邻居,那么节点 q 也必须是节点 p 的邻居。
必须将给定节点的拷贝作为对克隆图的引用返回。来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/clone-graph
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二、解题思路
dfs,深度优先遍历深度拷贝图,同时使用一map类型的词典save保存已经复制过的节点(防止进入圈内死循环),先复制当前节点,然后遍历当前节点的邻居,如果邻居节点已经存在save里,说明已经复制过,直接将save里的这个节点加入到当前复制节点的邻居队列里,然后遍历下一个邻居;如果不在save里,则复制当前邻居节点,加入到save里,同时加入到当前复制节点的邻居队列里,接着进入递归复制当前邻居的邻居节点,直到递归复制完所有邻居节点的邻居节点后,开始遍历当前节点的下一个邻居。直到所有节点都已经复制完,递归会结束。
三、代码
/*
// 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) {
map<Node *,Node *> save;
Node * new_node = new Node;
save[node] = new_node;
createNode(node,new_node,save);
return new_node;
}
void createNode(Node * node,Node* cnode,map<Node*,Node *> & save)
{
vector<Node*> & neighbors = node->neighbors;
vector<Node*> & new_nei = cnode->neighbors;
cnode->val = node->val;
for(int i = 0; i < neighbors.size(); i++)
{
if(save.count(neighbors[i]) > 0)
{
new_nei.push_back(save[neighbors[i]]);
}
else
{
Node * new_node = new Node;
save[neighbors[i]] = new_node;
createNode(neighbors[i],new_node,save);
new_nei.push_back(new_node);
}
}
}
};