leetcode 138. 复制带随机指针的链表

题目

难度：中等

思路

题目要求是复制原链表，需要另外新建链表，返回新建链表的头结点。
但原链表结构较为复杂，从左往右遍历原链表时，不能在复制节点时，事先知道当前节点的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;
	}
};