目录
题目:给你一个长度为 n 的链表,每个节点包含一个额外增加的随机指针 random ,该指针可以指向链表中的任何节点或空节点。
构造这个链表的 深拷贝。 深拷贝应该正好由 n 个 全新 节点组成,其中每个新节点的值都设为其对应的原节点的值。新节点的 next 指针和 random 指针也都应指向复制链表中的新节点,并使原链表和复制链表中的这些指针能够表示相同的链表状态。复制链表中的指针都不应指向原链表中的节点 。
链接:https://leetcode.cn/problems/copy-list-with-random-pointer
思路1(难实现)
看到这个题目,会先想到思路1,但是这种方法的复杂度是O(N^2),且在代码实现的层面来讲比较难搞定。
思路2(好理解,也简单些)
所以接下来介绍第二种方法。
第一步:是在OriginList处依次复制每个节点,遇到一个就malloc一个。
第二步:依次拷贝random指向
第三步:拆解!
具体要配合图片来看,更容易看明白。
具体实现代码如下
struct Node* copyRandomList(struct Node* head) {
//先拷贝一份节点
struct Node* curNode = head;
while(curNode)
{
//先保存OriginList中的下一个节点
struct Node* nextNode = curNode->next;
//malloc拷贝节点
struct Node* copyNode = (struct Node*)malloc(sizeof(struct Node));
//赋值
copyNode->val = curNode->val;
//更改next指向
curNode->next = copyNode;
copyNode->next = nextNode;
//curNode向后挪动,遍历OriginList
curNode = nextNode;
}
//依次拷贝random指向的节点
curNode = head;
while(curNode)
{
//保存CopyList中的节点
struct Node* copyNode = curNode->next;
if(curNode->random == NULL)
{
copyNode->random = NULL;
}
else
{
//让copyNode的random指向CopyList中的节点
copyNode->random = curNode->random->next;
}
//curNode要走到OriginList的下一个节点,走两步
curNode = curNode->next->next;
}
//拆解得到新的CopyList,把OriginList链接回去
curNode = head;
struct Node* copyHead = NULL;
struct Node* copyTail = NULL;
while(curNode)
{
//保存copyNode和nextNode
struct Node* copyNode = curNode->next;
struct Node* nextNode = copyNode->next;
//把原链表链接回去
curNode->next = nextNode;
//把拷贝节点尾插到CopyList
if(copyTail == NULL)
{
copyHead = copyTail = copyNode;
}
else
{
//尾插
copyTail->next = copyNode;
//更新尾节点
copyTail = copyNode;
}
//更新curNode在OriginList中的位置
curNode = nextNode;
}
return copyHead;
}