《剑指offer刷题笔记》36、复杂链表的复刻 【c++详细题解】

题目

请实现一个函数可以复制一个复杂链表。

在复杂链表中，每个结点除了有一个指针指向下一个结点外，还有一个额外的指针指向链表中的任意结点或者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;
    }
};