描述
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针random指向一个随机节点),请对此链表进行深拷贝,并返回拷贝后的头结点。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)。 下图是一个含有5个结点的复杂链表。图中实线箭头表示next指针,虚线箭头表示random指针。为简单起见,指向null的指针没有画出。
示例:
输入:{1,2,3,4,5,3,5,#,2,#}
输出:{1,2,3,4,5,3,5,#,2,#}
解析:我们将链表分为两段,前半部分{1,2,3,4,5}为ListNode,后半部分{3,5,#,2,#}是随机指针域表示。
以上示例前半部分可以表示链表为的ListNode:1->2->3->4->5
后半部分,3,5,#,2,#分别的表示为
1的位置指向3,2的位置指向5,3的位置指向null,4的位置指向2,5的位置指向null
如下图:
示例1
输入:{1,2,3,4,5,3,5,#,2,#}
返回值:{1,2,3,4,5,3,5,#,2,#}
题解:利用哈希表记录原节点和复制链表节点间映射(通过原节点直接找到对应的复制节点)
- 先建立next链表(不含random指向)
- 通过哈希表(遍历哈希表),根据原节点的random指向(有原节点和复制节点的映射)直接得到复制节点的random指向。
如图:建立好next指向的复制链表后,遍历哈希表。
原节点4的random指向2, 复制节点指向复制节点的2(即 m[4->random] ),因为原2节点和复制节点2映射。
class Solution {
public:
RandomListNode* Clone(RandomListNode* pHead) {
if(!pHead)return NULL;
unordered_map<RandomListNode*,RandomListNode*> m;
RandomListNode* head = pHead->next;
RandomListNode* hclone = new RandomListNode(pHead->label);
RandomListNode* tmp = hclone;
m[pHead] = hclone;
while(head){
RandomListNode* clone = new RandomListNode(head->label);
m[head] = clone;//节点和复制节点的映射。
hclone->next = clone;
hclone = hclone->next;
head = head->next;
}
for(auto &[key,value] : m){
value->random = (key->random == NULL ? NULL : m[key->random]);
}
return tmp;
}
};
运行时间:5ms
超过2.06% 用C++提交的代码
占用内存:552KB
超过64.67%用C++提交的代码