25、剑指offer--复杂链表的复制

题目描述

输入一个复杂链表（每个节点中有节点值，以及两个指针，一个指向下一个节点，另一个特殊指针指向任意一个节点），返回结果为复制后复杂链表的head。（注意，输出结果中请不要返回参数中的节点引用，否则判题程序会直接返回空）

 

思路：直接把复制的node放在原node的后面，这样结构变为：

上面为第一次遍历，第二次遍历时把红色的新node的random域赋值，规则是：

newNode->ranodm = oldNode->random->next;

然后第三次遍历把上面的链表拆分为两个即可。代码如下：

/**
 * Definition for singly-linked list with a random pointer.
 * struct RandomListNode {
 *     int label;
 *     RandomListNode *next, *random;
 *     RandomListNode(int x) : label(x), next(NULL), random(NULL) {}
 * };
 */
class Solution {
public:
    RandomListNode *copyRandomList(RandomListNode *head) {
        RandomListNode *newList = NULL;
        RandomListNode *newHead = NULL;
        RandomListNode *p = head;
 
        if(head==NULL)
            return NULL;
        //把每一个新的结点放入旧结点后面
        while(p!=NULL)
        {
            RandomListNode *q = new RandomListNode(p->label);
            q->next = p->next;
            p->next = q;
 
            p = q->next;
        }
 
        p = head;
        //在新结点中加入random
        while(p!=NULL)
        {
            if(p->random != NULL)
                p->next->random = p->random->next;
            p = p->next->next;//此时p需要跳两步才到旧的下一节点
        }
        p = head;
        //注意这里的条件，首先要判断p是否为null
        while(p!=NULL && p->next!=NULL)
        {
            if(p==head)
            {
                newHead = p->next;
                newList = newHead;
            }
            else
            {
                newList->next = p->next;
                newList = newList->next;
            }
            
            
             p->next = newList->next;  
             p = p->next;
             //如果使用p = newList->next;替换以上两步，则会改变原链表，错误
 
        }
 
        return newHead;//返回值是newHead而不是newList，因为此时newList指向尾部
    }
};