题目来源:牛客网
链接网址
题目描述
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),
返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
解题思路:
先保存一版正常链表
复制随机链表
使用递归和非递归的方法
/*
struct RandomListNode {
int label;
struct RandomListNode *next, *random;
RandomListNode(int x) :
label(x), next(NULL), random(NULL) {
}
};
*/
class Solution {
public:
RandomListNode* Clone(RandomListNode* pHead)
{
//RandomListNode* ret = make(pHead,mp);
map<RandomListNode*,RandomListNode*>mp;
if(pHead==nullptr)return nullptr;
RandomListNode* ret = new RandomListNode(pHead->label);
RandomListNode* q = ret;
mp[pHead]=ret;
for(auto p=pHead->next;p!=nullptr;q->next=new RandomListNode(p->label),mp[p]=q->next,p=p->next,q=q->next);
q->next=nullptr;
for(RandomListNode* p = pHead,*q=ret; p!=nullptr;p=p->next,q=q->next)
q->random = mp[p->random];
return ret;
}
RandomListNode* make(RandomListNode* root,map<RandomListNode*,RandomListNode*>&mp)
{
if(root==nullptr)return nullptr;
RandomListNode* ret = new RandomListNode(root->label);
mp[root] = ret;
ret->next = make(root->next,mp);
return ret;
}
};