复杂链表的复制
题目描述:
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
思路:在原链表的每一个节点的后面复制一个相同的节点,相连接,而原链表中每一个节点的随机指针的指向的节点的下一个节点就是复制链表节点的随机指针指向的节点。
代码:
class Solution {
public:
RandomListNode* Clone(RandomListNode* pHead)
{
if (pHead == nullptr)return nullptr;
RandomListNode*ptr = pHead;
while (ptr != nullptr){ //用于复制链表
RandomListNode*newptr = new RandomListNode(ptr->label);
newptr->next = ptr->next;
ptr->next = newptr;
ptr = newptr->next;
}
RandomListNode*ptr_1 = pHead;
RandomListNode*ptr_2 = ptr_1->next;
while (ptr_1 != nullptr){ //链接复制链表的随机指针
ptr_2 = ptr_1->next;
if (ptr_1->random){
ptr_2->random = ptr_1->random->next;
}
ptr_1 = ptr_2->next;
}
RandomListNode*head = pHead->next;
ptr = pHead; //该指针用于还原原来的链表
RandomListNode*ptr_V = head;
while (ptr_V->next != nullptr){ //抽出复制链表
ptr->next = ptr_V->next;
ptr = ptr->next;
ptr_V->next = ptr_V->next->next;
ptr_V = ptr_V->next;
}
ptr->next = nullptr;
return head;
}
};