给定一个单链表,链表除了包含next指针外,还包含一个random指针,该指针指向链表中某个结点。
请复制链表到一个新的链表,random指针需要指向新链表中对应的结点。比如原链表某个结点random指针指向第2个结点,那么新结点的random指针也要指向到新链表的第2个结点。
提示:此题存在空间复杂度O(1)的算法,不需要使用任何额外辅助空间。请不要改变原链表的结构。
哎,写代码差死了,根本没希望一次写对。各种遗漏各种笔误各种没想清楚。
/**
链表结点的定义(请不要在代码中定义该类型)
struct ListNode {
ListNode *next;
ListNode *random;
};
*/
// 返回复制的新链表头结点
ListNode* copyListWithRandomPtr(ListNode *head) {
if ( !head )
return head;
ListNode* pcur=head;
while(pcur)
{
ListNode* newAdd= new ListNode;
newAdd->next=pcur->next;
pcur->next=newAdd;
pcur=newAdd->next;
}
ListNode* pre=head;
while(pre)
{
pcur=pre->next;
pcur->random=pre->random->next;
pre=pcur->next;
}
ListNode guard;
ListNode* tail=&guard;
pre=head;
while(pre)
{
tail->next=pre->next;
tail=tail->next;
pre->next=pre->next->next;
pre=pre->next;
}
return guard.next;
}