剑指offer-复杂链表的复制

描述

输入一个复杂链表（每个节点中有节点值，以及两个指针，一个指向下一个节点，另一个特殊指针random指向一个随机节点），请对此链表进行深拷贝，并返回拷贝后的头结点。（注意，输出结果中请不要返回参数中的节点引用，否则判题程序会直接返回空）。 下图是一个含有5个结点的复杂链表。图中实线箭头表示next指针，虚线箭头表示random指针。为简单起见，指向null的指针没有画出。

 

 

求解思路：

1. 第一个循环先构建非随机链（即只考虑next指针），并用map记录label与新链表各节点的映射
2. 第二个循环根据map，完善随机指针。

代码：

 

/*
struct RandomListNode {
    int label;
    struct RandomListNode *next, *random;
    RandomListNode(int x) :
            label(x), next(NULL), random(NULL) {
    }
};
*/
class Solution {
public:
    RandomListNode* Clone(RandomListNode* pHead) {
        //  如果把复杂链表看作随机链和非随机链，那先确定非随机链
        // 然后依次检验每个节点的随机指针，读取val，通过val确定新链的随机指针
        // 改进：在构建非随机链表时，新建map映射 val和random(这里记录节点)
        std::map<int,RandomListNode*> lm;
        RandomListNode* newHead=NULL;
        RandomListNode* tail;   // 记录尾节点
        // 完成非随机链表的构建
        RandomListNode* ergNode=pHead;
        while(ergNode!=nullptr){
            RandomListNode* rn=new RandomListNode(ergNode->label);
            if(newHead==nullptr){
                newHead=rn;
            }else{
                tail->next=rn;
            }
            tail=rn;
            // 构建label和node的映射
            // lm.insert(pair<int,RandomListNode*>(rn->label,rn));
            lm[rn->label]=rn;
            ergNode=ergNode->next;
        }
        // 再用一个循环完成随机链表的构建
        ergNode=pHead;  // 遍历原有链表
        RandomListNode* repNode=newHead;  // 补充现有链表
        while(ergNode!=nullptr){
            if(ergNode->random!=nullptr){
                repNode->random=lm[ergNode->random->label];
                int f=1;
            }
            ergNode=ergNode->next;
            repNode=repNode->next;
        }
        return newHead;
    }
};