题目
请实现一个函数可以复制一个复杂链表。
在复杂链表中,每个结点除了有一个指针指向下一个结点外,还有一个额外的指针指向链表中的任意结点或者null。
注意:
- 函数结束后原链表要与输入时保持一致。
思路
原链表图形: 虚线为random
指针
1、在每个节点的后面加上它的复刻,将原链表和复刻链表连在一起。
2、从前往后遍历每一个原链表节点,对于有random
指针的节点p
,我们让它的p->next->random = p->random->next
,这样我们就完成了对原链表random
指针的复刻。
3、最后我们把原链表和复刻链表拆分出来,并将原链表复原。
代码
/**
* 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的下一个节点
p->next = np;//将原链表当前节点p指向复刻节点
np->next = next;//复刻节点np指向原链表当前节点的下一个节点
p = next; //p指针后移
}
for (auto p = head; p; p = p->next->next)//对原链表random指针的复刻。
{
if (p->random)
p->next->random = p->random->next;
}
auto dummy = new ListNode(-1); //虚拟头结点
auto cur = dummy; //尾节点
for (auto p = head; p; p = p->next) //将原链表和复刻链表拆分出来,并将原链表复原。
{
cur->next = p->next;
cur = cur->next;
p->next = p->next->next;
}
return dummy->next;
}
};