题目:
请实现 copyRandomList
函数,复制一个复杂链表。在复杂链表中,每个节点除了有一个 next
指针指向下一个节点,还有一个 random
指针指向链表中的任意节点或者 null
。
解题思路:
这题真是,一开始连题目都看不明白的。一开始想的是就循环遍历new新结点。代码如下:
//大体的思路
Node* copyRandomList(Node* head){
auto cur=head;
auto tmp=cur;
while(cur){
auto node=new Node(cur->val);
node->random=cur->random;
node->next=cur->next;
cur=cur->next;
}
}
是很容易想到了这种写法。。后面才知道这题是想要深拷贝的意思。就是newnode的next也要是新创建出来的node。把新结点的next=原先的结点的next,这是错误的。
auto newnode1=new Node(1);
auto newnode2=new Node(2);
newnode1->next=newnode2;
//newnode1->next=cur->next; 这种是不行的,这种 next就是原来的结点的
题目是想要这种的效果。所以当我们new出一个新结点后,该结点的next不能是原先结点的next,需要是又一个新结点。
所以,我们需要保存到每个新结点。那新结点和原先的结点就要一一对应。那可以用std::unordered_map。
来看看代码
class Solution {
public:
Node* copyRandomList(Node* head) {
if(!head)
return head;
unordered_map<Node*,Node*> map; //<原结点,新结点>
auto cur=head;
while(cur){
map.emplace(cur,new Node(cur->val));
cur=cur->next;
}
cur=head;
while(cur){
// map[cur]->next=cur->next; //这样写就是新结点->next=旧结点->next,不行的,是要深拷贝的意思
map[cur]->next=map[cur->next]; //新结点的next=新结点的下一个结点
map[cur]->random=map[cur->random];
cur=cur->next;
}
return map[head];
}
};
这题需要理解的是需要创建出一个链表,该链表的结点都是new出来的,而其结点的next和random对应的也是new出来的结点。