[剑指 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; } };