复制带随机指针的链表
-
题目描述:
给定一个链表,每个节点包含一个额外增加的随机指针,该指针可以指向链表中的任何节点或空节点。
要求返回这个链表的 深拷贝。
我们用一个由 n
个节点组成的链表来表示输入/输出中的链表。每个节点用一个 [val, random_index]
表示:
val
:一个表示Node.val
的整数。random_index
:随机指针指向的节点索引(范围从0
到n-1
);如果不指向任何节点,则为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。
-
解题思路:
(1)拷贝数据
创建新节点,在当前节点后插入该新节点,新节点的数据来自于当前节点的拷贝
直接看代码,思路清晰:
struct Node* cur=head;
while(cur)
{
//(1)拷贝数据
struct Node* newnode=(struct Node*)malloc(sizeof(struct Node));
newnode->val=cur->val;
newnode->next=cur->next;
cur->next=newnode;
cur=newnode->next;
}
(2)拷贝随机指针
由于第(1)步操作,每个节点的拷贝节点尾插在当前节点后,在遍历找拷贝random随机指针的时候如果random不为null,可通过当前random指针的next找到
看代码如下:
//(2)拷贝随机指针
cur=head;
while(cur)
{
struct Node* copy=cur->next;
if(cur->random)
{
copy->random=cur->random->next;
}
else
copy->random=NULL;
cur=copy->next;
}
(3)将拷贝后的链表与原链表进行拆分
为方便找到random随机指针,将copy的链表与原链表连接在一起,然后进行拆分
代码十分简洁,一看就懂,创建cur,copy,next指针,同时创建newhead指针保存拆分后拷贝链表
(3)链表拆分
cur=head;
struct Node* newhead=cur->next;
while(cur)
{
struct Node* copy=cur->next;
struct Node* next=copy->next;
cur->next=next;
if(next)
copy->next=next->next;
cur=next;
}
return newhead;