复杂链表的复制

题目描述

输入一个复杂链表（每个节点中有节点值，以及两个指针，一个指向下一个节点，另一个特殊指针指向任意一个节点）。

 

/*
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];
    }
};