题目
难度:中等
思路
题目要求是复制原链表,需要另外新建链表,返回新建链表的头结点。
但原链表结构较为复杂,从左往右遍历原链表时,不能在复制节点时,事先知道当前节点的next和random域,所以可以先复制节点并存储下来,在后续遍历过程中再补齐该节点的next和radom域。
存储时利用哈希map<原链表节点,复制节点>,遍历原链表过程中,利用原链表节点去map中查看,前文遍历时是否复制过该节点:没有,则创建复制节点,存入map;有,则直接更新复制节点对应域。
c++代码
class Solution {
public:
Node* copyRandomList(Node* head) {
unordered_map<Node*, Node*>mp; //<原节点,复制的节点>
if (head == NULL) return NULL;
Node* new_head = new Node(head->val);
mp[head] = new_head; //原头结点,复制头结点
mp[NULL] = NULL;
Node* tmp = head;
while (tmp) {
if (tmp->next&&mp.find(tmp->next) == mp.end()) { //tmp的next不在map中,即该节点还未被复制
Node* new_node = new Node(tmp->next->val);
mp[tmp->next] = new_node;
}
mp[tmp]->next = mp[tmp->next]; //修改复制后的tmp->next的值
if (tmp->random&&mp.find(tmp->random) == mp.end()) { //tmp的random不在map中,即该节点还未被复制
Node* new_node = new Node(tmp->random->val);
mp[tmp->random] = new_node;
}
mp[tmp]->random = mp[tmp->random]; //修改复制后的tmp->random的值
tmp = tmp->next; //后移
}
return new_head;
}
};