给定一个链表,每个节点包含一个额外增加的随机指针,该指针可以指向链表中的任何节点或空节点。
要求返回这个链表的深拷贝。
示例:
输入:
{“KaTeX parse error: Expected '}', got 'EOF' at end of input: …":"1","next":{"id”:“2”,“next”:null,“random”:{“KaTeX parse error: Expected 'EOF', got '}' at position 9: ref":"2"}̲,"val":2},"rand…ref”:“2”},“val”:1}
解释:
节点 1 的值是 1,它的下一个指针和随机指针都指向节点 2 。
节点 2 的值是 2,它的下一个指针指向 null,随机指针指向它自己。
提示:
你必须返回给定头的拷贝作为对克隆列表的引用。
思路
该题要求返回一个带随机指针的链表的拷贝,并且不能修改原链表结构。
拷贝一个链表很简单,但是问题是怎么拷贝拷贝后链表的随机指针。我第一个想法是用map,将每个node的随机指针指向存在map里,例如key存node的索引位置,value存随机指针指向的位置。但是这样每次还要根据索引位置找到随机指针指向的node地址,空间和实践复杂度都高。
一个巧妙的做法是将每个结点的拷贝结点跟在原结点的后面,这样我们可以方便设置拷贝后结点的随机指针指向。
/*
// Definition for a Node.
class Node {
public:
int val;
Node* next;
Node* random;
Node() {}
Node(int _val, Node* _next, Node* _random) {
val = _val;
next = _next;
random = _random;
}
};
*/
class Solution {
public:
Node* copyRandomList(Node* head) {
if (!head) return NULL;
Node* cur=head,*next=head->next;
while(cur)
{
Node* node=new Node(cur->val,NULL,NULL);
next=cur->next;
cur->next=node;
node->next=next;
cur=next;
}
//cout<<"111"<<endl;
cur=head;
while(cur)
{
if(cur->random)
{
cur->next->random=cur->random->next;
}
cur=cur->next->next;
}
//cout<<"222"<<endl;
Node* new_list=head->next,*n1=head,*n2=new_list;
while(n1)
{
n1->next=n2->next;
if (n1->next)
n2->next=n2->next->next;
else
n2->next=NULL;
n1=n1->next;
n2=n2->next;
}
//printList(new_list);
//cout<<endl;
//printList(head);
//cout<<"aaaaaaaa";
return new_list;
}
void printList(Node* head)
{
Node* node=head;
while(node)
{
cout<<node->val<<",";
node=node->next;
}
}
};