1,题目要求
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
2,题目思路
对于这道题,要求对一个含有随机指针的链表进行拷贝。
在问题的解决上,我们使用map来解决问题:
其中,key是原始链表中的节点,而value是新链表中的节点。
如果mp[pHead]非空,说明当前节点已经被拷贝了(即前面有random创建了节点),直接返回该节点。
而如果mp[pHead]为空,则对该节点进行创建——label、next以及random。
3,代码实现
/*
struct RandomListNode {
int label;
struct RandomListNode *next, *random;
RandomListNode(int x) :
label(x), next(NULL), random(NULL) {
}
};
*/
class Solution {
public:
RandomListNode* Clone(RandomListNode* pHead){
if(!pHead)
return nullptr;
if(mp[pHead]!=nullptr)
return mp[pHead];
mp[pHead] = new RandomListNode(pHead->label);
mp[pHead]->next = Clone(pHead->next);
mp[pHead]->random = Clone(pHead->random);
return mp[pHead];
}
private:
unordered_map<RandomListNode*, RandomListNode*> mp;
};