刷題筆記（十二）——複雜鏈表複製

刷題筆記（十二）——複雜鏈表複製

題目描述

輸入一個複雜鏈表（每個節點中有節點值，以及兩個指針，一個指向下一個節點，另一個特殊指針指向任意一個節點），返回結果爲複製後複雜鏈表的head。（注意，輸出結果中請不要返回參數中的節點引用，否則判題程序會直接返回空）

思路：開始沒大看懂題啥意思，看了給定的結構體才明白。

首先將各個節點進行復制A——>.A*，並將複製的節點A*插在A、B之間。

得到鏈表爲：A、A*、B、B*、C、C*......

修改next指針和random指針，A*的next指針指向next的next指針；

拆分，修改next指針

/*
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) return NULL;
        RandomListNode* p=pHead;
        while(p!=NULL)
        {
            RandomListNode *node = new RandomListNode(p->label);
            node->next=p->next;
            p->next=node;
            p=node->next;
        }
        p=pHead;
        while(p!=NULL)
        {
            RandomListNode *node = p->next;
            if(p->random){               
                node->random = p->random->next;
            }
            p = node->next;
            /*if(p->random->next!=NULL)
                p->next->random=p->random->next;
            p=p->next->next;*/
        }
        p=pHead;
        RandomListNode* q=p->next;
        RandomListNode *tmp;
        while(p->next!=NULL)
        {
            tmp = p->next;
            p->next =tmp->next;
            p = tmp;
        }
        return q;
    }
};