给定一个链表,每个节点包含一个额外增加的随机指针,该指针可以指向链表中的任何节点或空节点。
要求返回这个链表的深拷贝。
示例:
输入:
{"$id":"1","next":{"$id":"2","next":null,"random":{"$ref":"2"},"val":2},"random":{"$ref":"2"},"val":1}
解释:
节点 1 的值是 1,它的下一个指针和随机指针都指向节点 2 。
节点 2 的值是 2,它的下一个指针指向 null,随机指针指向它自己。
提示:
你必须返回给定头的拷贝作为对克隆列表的引用。
这个题目和133克隆图的题目就是一样的思想,一模一样。。。。。直接设置一个哈希表存放已经遍历过的节点,键存放源列表节点位置,值存放克隆列表节点位置。深度优先搜索即可。
C++源代码:
/*
// Definition for a Node.
class Node {
public:
int val;
Node* next;
Node* random;
Node() {}
Node(int _val, Node* _next, Node* _random) {
val = _val;
next = _next;
random = _random;
}
};
*/
class Solution {
public:
Node* copyRandomList(Node* head) {
unordered_map<Node*, Node*> m;
return clone(head, m);
}
Node* clone(Node* node, unordered_map<Node*, Node*>& m){
if (!node) return node;
if (m.count(node)) return m[node];
Node *newNode = new Node(node->val, NULL, NULL);
m[node] = newNode;
newNode->next = clone(node->next, m);
newNode->random = clone(node->random, m);
return newNode;
}
};
python3源代码:
"""
# Definition for a Node.
class Node:
def __init__(self, val, next, random):
self.val = val
self.next = next
self.random = random
"""
class Solution:
def copyRandomList(self, head: 'Node') -> 'Node':
m = {}
return self.clone(head, m)
def clone(self, node, m):
if node==None: return node
if node in m.keys(): return m[node]
newNode = Node(node.val, None, None)
m[node] = newNode
newNode.next = self.clone(node.next, m)
newNode.random = self.clone(node.random, m)
return newNode