Leetcode-138. Copy List with Random Pointer

题目

A linked list is given such that each node contains an additional random pointer which could point to any node in the list or null.

Return a deep copy of the list.
深拷贝一个链表，链表除了含有next指针外，还包含一个random指针，该指针指向字符串中的某个节点或者为空。
拷贝原链表

思路

第一步：在原链表中创建每一个节点的复制节点，所有值均相同
第二步：修改每一个复制节点的random的值，如果不为空，random的值为random节点的next节点
第三步：拆分链表，复原原链表，得到拷贝链表

代码

class Solution {
public:
    RandomListNode *copyRandomList(RandomListNode *head) {
        if (head == NULL) return NULL;

        ///假设：l1代表原链表中的节点；l2代表新链表中的节点
        RandomListNode *new_head, *l1, *l2;

        ///第一步：在每一个l1后面创建一个l2，并让l1指向l2，l2指向下一个l1；
        for (l1 = head; l1 != NULL; l1 = l1->next->next) {
            l2 = new RandomListNode(l1->label);
            l2->next = l1->next;
            l2->random = l1->random;
            l1->next = l2;
        }

        ///第二步：给l2的random赋值，l1的random的next指向的就是l2的random的目标；
        new_head = head->next;
        for (l1 = head; l1 != NULL; l1 = l1->next->next) {
            if (l1->random != NULL) l1->next->random = l1->random->next;
        }

        ///第三步：需要将整个链表拆成两个链表，具体做法是让l1
        ///的next指向后面的后面;l2的next也指向后面的后面。
        for (l1 = head; l1 != NULL; l1 = l1->next) {
            l2 = l1->next;
            l1->next = l2->next;
            if (l2->next != NULL)
                l2->next = l2->next->next;
        }
        return new_head;
    }
};