题目描述:
请实现一个函数可以复制一个复杂链表。在复杂链表中,每个结点除了有一个指针指向下一个结点外,还有一个额外的指针指向链表中的任意结点或者null。
分析:
可以用哈希表来将原链表来映射到新的链表,也可以不用hash,采用一种新的方式解决。
设原链表为abcdef,我们在每个结点后面插入新的结点,其值和前面结点一致。得到aAbBcCdDeEfF。然后比如a的random指向d,我们就把A的random指向d的后一个元素也就是D。最后去掉多余节点,可采用新构建一个头结点也可以直接用a,将A的后继改为B,B的改为C...。到最后得到ABCDEF,也就是我们所需要求得复制的新链表。
/**
* Definition for singly-linked list with a random pointer.
* struct ListNode {
* int val;
* ListNode *next, *random;
* ListNode(int x) : val(x), next(NULL), random(NULL) {}
* };
*/
class Solution {
public:
ListNode *copyRandomList(ListNode *head) {
for(auto p = head;p;){
auto np = new ListNode(p->val);
auto next = p->next;
p->next = np;
np->next = next;
p = next;
}
for(auto p = head;p;p = p->next->next){
if(p->random) p->next->random = p->random->next;
}
auto d = new ListNode(-1);
auto cur = d;
for(auto p = head;p;p = p->next){
cur->next = p->next;
cur = cur->next;
p = p->next;
}
return d->next;
}
};