思路:
1.将每个节点复制一次,放在原节点和原节点后面一个节点之间
2.给每个复制节点的random指针指向原节点的指针对应random指向节点的下一个节点
3.分离链表
这样是没有占用额外空间并且时间复杂度是O(n),代码如下所示:
class Solution {
public:
Node* copyRandomList(Node* head) {
cloneNodes(head);
setRandomPtr(head);
return splitList(head);
}
//复制每个节点 复制节点放在原节点和原节点后面节点之间
void cloneNodes(Node* head)
{
Node* p=head;
while(p!=NULL)
{
Node* node=new Node(p->val);
node->next=p->next;
p->next=node;
p=node->next;
}
}
//设置复制节点的random指针
void setRandomPtr(Node* head)
{
Node* p=head;
while(p!=NULL)
{
Node* pNext=p->next;
if(p->random)
{
pNext->random=p->random->next;
}
p=pNext->next;
}
}
//拆分链表
Node* splitList(Node* head)
{
Node* p=head;
Node* pClone=NULL;
Node* pCloneHead=NULL;
if(p!=NULL)
{
pCloneHead=pClone=p->next;
}
else return NULL;//解决原链表头节点为空的情况
while(pClone->next!=NULL)
{
p->next=p->next->next;
pClone->next=pClone->next->next;
p=p->next;
pClone=pClone->next;
}
p->next=NULL;//原链表的尾指针需要更新 容易出错
return pCloneHead;
}
};
其余思路的思考:
1.先复制所有节点一遍并且直接用next指针先把复制节点连起来,再遍历一次,在遍历过程中更新复制节点的random指针,由于本来就是两个链表所以不需要把链表拆分。在更新复制链表的random指针时由于需要根据原链表的位置来遍历复制的链表,时间复杂度是O(n^2)
2.可以用空间换时间,1中复制复杂链表的问题在于,random的更新,需要遍历原链表找到位置,再遍历当前复制链表找到random该指向的节点,如果想快一点找到,容易想到,直接建立一个哈希表,保存原节点和复制节点的映射关系,这样可以用两个指针同步遍历一遍两个链表就可以完成random的设置