[剑指 Offer]复杂链表的复制

[剑指 Offer]复杂链表的复制

剑指 Offer-复杂链表的复制

题目描述

请实现 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 。

解题思路

• 每个节点的后面插入复制的节点，如 1 -> 2 -> 3 ->4 -> null 复制完就是 1 -> 1 -> 2 -> 2 -> 3 - > 3 -> 4 ->4 -> null
• 给新插入的节点的random赋值
• 拆分

实现代码

class Solution {
public:
    Node* copyRandomList(Node* head) {   
        if(head==NULL)
            return NULL;    
        //插入新节点 1->2->3->null  ——> 1->1->2->2->3->3->4->4->null
        for(Node* p=head;p!=NULL;p=p->next->next){
            Node* temp=new Node(p->val);
            temp->next=p->next;            
            p->next=temp;
        }
        //建立新节点的random
        for(Node* p=head;p!=NULL;p=p->next->next){
            if(p->random==NULL)
                p->next->random=NULL;
            else
                p->next->random=p->random->next;
        }
        //拆分
        Node* p=head;
        Node* copyhead=p->next;
        Node* q=p->next;
        while(1){
            p->next=q->next;
            p=p->next;
            if(p==NULL)
                break;
            q->next=p->next;
            q=q->next;
        }
        return copyhead;
    }
};