题目
A linked list is given such that each node contains an additional random pointer which could point to any node in the list or null.
Return a deep copy of the list.
分析
这道题就是深度复制一个链表,这个链表和其他的区别之处在于每个Node除了指向下一个Node之外,还可以指向一个随机的Node,这个随机的Node可以指向任意Node或者NULL,下图是示例,节点3随机指向节点6,节点4随机指向了节点2,节点5随机指向了节点3。
首先,我们在不管指向随机Node这一部分,复制一个链表是很简单。
RandomListNode* result = new RandomListNode(0);
RandomListNode* tail = result;
while(head!=NULL){
tail->next = new RandomListNode(tmp->label);
tail = tail->next;
head= head->next;
}
下面怎么处理复制随机指向节点问题了。
对于示例的链表来说就是复制:节点3随机指向节点6,节点4随机指向了节点2,节点5随机指向了节点3这些指向。
可是怎么做到呢?建立新链表和旧链表Node之间的对应关系,hash表就可以轻松做到这一点,比如,原链表的节点4随机指向节点2,那么在扫描原链表的时候,就可以知道新链表的节点4’先指向原链表的节点2,通过hash列表可以由原列表的节点2指向新链表的2’,从而实现随机指向的复制。直接说的话不好理解,看下面一张图你就明白了。
所以综合起来就是两次扫描原链表,第一次扫描复制next关系,将主链表复制,第二次扫描通过Map复制random关系,完成整个链表的深度复制。
代码
/**
* Definition for singly-linked list with a random pointer.
* struct RandomListNode {
* int label;
* RandomListNode *next, *random;
* RandomListNode(int x) : label(x), next(NULL), random(NULL) {}
* };
*/
class Solution {
public:
RandomListNode *copyRandomList(RandomListNode *head) {
RandomListNode* result = new RandomListNode(0);
RandomListNode* tail = result;
RandomListNode* tmp = head;
unordered_map<RandomListNode*,RandomListNode*> mapping;
if(head==NULL){
return head;
}
while(tmp!=NULL){
tail->next = new RandomListNode(tmp->label);
tail = tail->next;
mapping[tmp]=tail;
tmp = tmp->next;
}
tail = result;
tail = tail->next;
tmp = head;
while(tmp!=NULL){
if(tmp->random==NULL){
tail->random = NULL;
}
else{
tail->random = mapping[tmp->random];
}
tmp = tmp->next;
tail = tail->next;
}
return result->next;
}
};