复杂链表的复制

题目描述

/*
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 == NULL)
return NULL;

RandomListNode* p = pHead;
std::map<int,RandomListNode*> mm;
while(p!= NULL)
{
RandomListNode* ptem = new RandomListNode(p->label);
mm[p->label] = ptem;
p = p->next;
}
p = pHead;
while(p != NULL)
{
if(p->next != NULL)
mm[p->label]->next = mm[p->next->label];
if(p->random != NULL)
mm[p->label]->random = mm[p->random->label];
p = p->next;
}
return mm[pHead->label];
}
};