LeetCode——剑指 Offer 35. 复杂链表的复制

一、题目

请实现 copyRandomList 函数，复制一个复杂链表。在复杂链表中，每个节点除了有一个 next 指针指向下一个节点，还有一个 random 指针指向链表中的任意节点或者 null。

示例 1：

输入：head = [[7,null],[13,0],[11,4],[10,2],[1,0]]
输出：[[7,null],[13,0],[11,4],[10,2],[1,0]]
示例 2：

输入：head = [[1,1],[2,1]]
输出：[[1,1],[2,1]]
示例 3：
输入：head = [[3,null],[3,0],[3,null]]
输出：[[3,null],[3,0],[3,null]]
示例 4：
输入：head = []
输出：[]
解释：给定的链表为空（空指针），因此返回 null。
提示：
-10000 <= Node.val <= 10000
Node.random 为空（null）或指向链表中的节点。
节点数目不超过 1000 。

二、思路

刚开始我也没有想到很好的方法，甚至把题目看错了，题目给出的random是Node类型的，我傻乎乎的以为是int类型，拿到题目就用数组开始莽，写着发现不对劲，后来去评论区看了一下大佬的思路，于是我开始按照以下的思路开始写
在给出的链表中，把每个节点挨着复制一遍（维护next域）
比如：原来的链表为1->2->3->NULL，复制完成后的链表为1->1->2->2->3->3->NULL
然后再把random域给维护一下，即 node->next->random=node->random->next（这里看不懂的可以画图看一下），需要注意的是random为NULL的情况（在下面的代码中会有所体现）
最后将链表一分为二即可

三、C++解法

/*
// Definition for a Node.
class Node {
public:
    int val;
    Node* next;
    Node* random;
    
    Node(int _val) {
        val = _val;
        next = NULL;
        random = NULL;
    }
};
*/
class Solution {
public:
	//指定节点后插节点
    void insertNext(struct Node* node){
        struct Node *p = (struct Node *)malloc(sizeof(Node));
        p->val=node->val;
        p->random=NULL;
        p->next=node->next;
        node->next=p;
    }

    Node* copyRandomList(Node* head) {
        if(head == NULL) return NULL;
        Node *p=head,*copyhead,*index;
        //维护next域
        while(p!=NULL){
            insertNext(p);
            p=p->next->next;
        }
        p=head;
        //维护random域
        while(p!=NULL){
            if(p->random!=NULL)
            p->next->random=p->random->next;
            p=p->next->next;
        }
        p= head;
        //链表一分为二
        copyhead = p->next;
        index = copyhead;
        while(index->next!=NULL){
            p->next=p->next->next;
            index->next=p->next->next;
            p=p->next;
            index = index->next;
        }
        p->next = NULL;
        return copyhead;
    }
};