[剑指offer]复杂链表的复刻

复杂链表的复刻

题目

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

在复杂链表中，每个结点除了有一个指针指向下一个结点外，还有一个额外的指针指向链表中的任意结点或者null。

注意

• 函数结束后原链表要与输入时保持一致。

样例

输入：[[1, 1], [0, 2], [1, 4], [3, -1], [3, 2]]

输出：[[1, 1], [0, 2], [1, 4], [3, -1], [3, 2]]

题解

由于不能直接返回原链表，所以我们需要构造出一个新的链表

思路

• 在原链表的基础上，复制每个结点到原结点的后边
• 使新结点和原结点的 random 结点相同
• 最后拆解出两个相同的链表

/**
 * 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; p = p->next->next) {
            auto np = new ListNode(p->val);
            np->next = p->next;
            p->next = np;
        }
        
        for (auto p = head; p; p = p->next->next) 
            if (p->random) p->next->random = p->random->next;
            
        ListNode* dummy = new ListNode(-1);
        for (auto p = head, q = dummy; p; p = p->next, q = q->next) {
            q->next = p->next;
            p->next = p->next->next;
        }

        return dummy->next;
    }
};