剑指offer-复杂链表的复制

题目描述

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

思路

• 先复制对应的节点并连接在对应节点的后面
• 设置各个节点的random指针
• 将新旧链表分割开来（这里我是一脸蒙蔽，明明我返回写的函数没有返回引用，就是通过不了测试…）
• 时间复杂度为$O(n)$，空间复杂度为$O(1)$

AC代码

#include <iostream>
using namespace std;

struct RandomListNode
{
    int label;
    struct RandomListNode *next, *random;
    RandomListNode(int x) : label(x), next(NULL), random(NULL)
    {
    }
};

class Solution
{
public:
    RandomListNode *Clone(RandomListNode *pHead)
    {
        first(pHead);
        second(pHead);
        return third(pHead);
    }
    void first(RandomListNode *pHead) //复制每个节点，并将节点复制到对应节点的后面
    {
        RandomListNode *p = pHead;
        while (p)
        {
            RandomListNode *new_node = new RandomListNode(p->label); //创建新节点
            //将新节点复制到原节点的后面
            new_node->next = p->next; //这里不能复制节点的random指针，因为后面的新节点还没有创建，就不能连接在一起
            new_node->random = NULL;  //先设置为NULL
            p->next = new_node;

            //指针移动到新的节点上
            p = new_node->next;
        }
    }
    void second(RandomListNode *pHead) //连接每个节点的random指针
    {
        RandomListNode *p = pHead;
        while (p)
        {
            RandomListNode * temp = p->next;
            if (p->random)
                temp->random = p->random->next;
            p = temp->next;
        }
    }
    RandomListNode *third(RandomListNode *pHead) //将新旧两个链表分割出来
    {
        /* 我这自己写的就是通过不了测试，感觉也没问题啊....
        if (pHead == NULL)
            return NULL;
        RandomListNode * curnode = pHead->next;
        RandomListNode * res = curnode;
        RandomListNode * temp;
        while (curnode->next)
        {
            temp = curnode->next;
            curnode->next = temp->next;
            curnode = temp->next;
        }
        return res;
        */

        if (pHead == NULL)
            return NULL;
        RandomListNode * pnew = pHead->next;
        RandomListNode *temp;
        RandomListNode * curnode = pHead;
        while(curnode->next)   //将旧链表重新连接好，这样也就是将新旧链表分割开来了
        {
            temp = curnode->next;
            curnode->next =temp->next;
            curnode = temp;
        }
        return pnew;
    }
};
int main()
{
    RandomListNode *head = new RandomListNode(1);
    RandomListNode *p1 = new RandomListNode(2);
    RandomListNode *p2 = new RandomListNode(3);
    head->next = p1;
    p1->next = p2;
    head->random = p2;
    p2->random = p1;
    Solution so;
    RandomListNode *pnew = so.Clone(head);
    while (pnew)
    {
        cout << pnew->label << "  ";
        pnew = pnew->random;
    }
    cout << endl;

    return 0;
}