题目:
给你一个长度为 n 的链表,每个节点包含一个额外增加的随机指针 random ,该指针可以指向链表中的任何节点或空节点。
构造这个链表的 深拷贝。 深拷贝应该正好由 n 个 全新 节点组成,其中每个新节点的值都设为其对应的原节点的值。新节点的 next 指针和 random 指针也都应指向复制链表中的新节点,并使原链表和复制链表中的这些指针能够表示相同的链表状态。复制链表中的指针都不应指向原链表中的节点 。
例如,如果原链表中有 X 和 Y 两个节点,其中 X.random --> Y 。那么在复制链表中对应的两个节点 x 和 y ,同样有 x.random --> y 。
返回复制链表的头节点。
用一个由 n 个节点组成的链表来表示输入/输出中的链表。每个节点用一个 [val, random_index] 表示:
val:一个表示 Node.val 的整数。
random_index:随机指针指向的节点索引(范围从 0 到 n-1);如果不指向任何节点,则为 null 。
你的代码 只 接受原链表的头节点 head 作为传入参数。
解题思路:
首先我们复制这个链表要考虑的两个问题
1.复制它的值,复制值简单,新建一个链表把值全部拷贝进去即可
2.复制它的指针,next指针可以很轻松的找到它的下一个地址。但是random指针拷贝就比较困难了,他需要找到之前链表random指向的位置相对random位置的距离,如下图:
但是这样实现,用代码写起来比较困难,而且时间复杂度也是O(N^2)
容易理解的方法
我们让拷贝的每个新节点跟在旧节点的后面,例如:
图中绿色的线连起来就是新的链表
第一步
我们把新链表连接起来
struct Node
{
int val;
struct Node *next;
struct Node *random;
};
struct Node* copyRandomList(struct Node* head)
{
//把新链表连接起来
struct Node* cur = head;
while(cur)
{
struct Node* copy = (struct Node*)malloc(sizeof(struct Node));
copy->val = cur->val;
copy->next = cur->next;
cur->next = copy;
cur = copy->next;
}
}
第二步(重点)
使拷贝的random指向拷贝链表中它所指向的位置,我们只需要让拷贝的random指向它前一个值的random的next(看图理解,如下图)
如果拷贝链表前面的random指想NULL,那拷贝链表的random也指向空
cur = head;
while(cur)
{
struct Node* copy = cur->next;
struct Node* next = copy->next;
if(cur->random == NULL)
{
copy->random = NULL;
}
else
{
copy->random = cur->random->next;
}
cur = next;
}
第三步
拆出新链表,还原原链表
cur = head;
//创建新链表的头和尾
struct Node* newhead = NULL;
struct Node* tail = NULL;
while(cur)
{
struct Node* copy = cur->next;
struct Node* next = copy->next;
if(newhead == NULL)
{
newhead = tail = copy;
}
else
{
tail->next = copy;
tail = tail->next;
}
//还原原来链表的指向
cur->next = next;
cur = next;
}