请实现copyRandomList函数,复制一个复杂链表。在复杂链表中,每个节点除了有一个next指针指向下一个节点,还有一个random指针指向链表中的任意节点或者null。
输入:head = [[7,null],[13,0],[11,4],[10,2],[1,0]]
输出:[[7,null],[13,0],[11,4],[10,2],[1,0]]
思路:哈希表
复制普通链表很简单,只需遍历链表,每轮建立新节点 + 构建前驱结点pre和当前节点node的引用指向即可。
我们用哈希表来解决这个问题:
首先创建一个哈希表,再遍历原链表,遍历的同时再不断创建新节点
我们将原节点作为key,新节点作为value放入哈希表中。
第二步我们再遍历原链表,这次我们要将新链表的next和random指针给设置上。
从上图中,可以发现,原节点和新节点是一一对应关系,所以,
- map[cur->val],得到的就是对应的新节点
- map[cur->next],得到的就是对应的新节点.next
- map[cur->random],得到的就是对应的新节点.random
所以,我们只需要再次遍历原链表,然后设置:
新节点.next -> map[cur->next]
新节点.random -> map[cur->random]
最后,map[head],也就是对应的新链表的头结点返回。
利用哈希表的查询特点,考虑构建原链表节点和新链表对应节点的键值对映射关系,再遍历构建新链表各节点的next和random引用指向即可。
算法流程:
- 若头节点head为空节点,直接返回null
- 初始化: 哈希表dic,节点cur指向头节点
- 复制链表:
- 构建新链表的引用指向:
- 返回值:新链表的头节点dic[cur];
/*
// Definition for a Node.
class Node {
public:
int val;
Node* next;
Node* random;
Node(int _val) {
val = _val;
next = NULL;
random = NULL;
}
};
*/
class Solution {
public:
Node* copyRandomList(Node* head) {
if(head == nullptr) return nullptr;
Node* cur = head;
unordered_map<Node*, Node*> map;
// 3. 复制各节点,并建立 “原节点 -> 新节点” 的 Map 映射
while(cur != nullptr) {
map[cur] = new Node(cur->val);
cur = cur->next;
}
cur = head;
// 4. 构建新链表的 next 和 random 指向
while(cur != nullptr) {
map[cur]->next = map[cur->next];
map[cur]->random = map[cur->random];
cur = cur->next;
}
// 5. 返回新链表的头节点
return map[head];
}
};