【LeetCode练习题】Copy List with Random Pointer

Copy List with Random Pointer

A linked list is given such that each node contains an additional random pointer which could point to any node in the list or null.

Return a deep copy of the list.

 

题目意思：

深拷贝一个给定的带random指针的链表，这个random指针可能会指向其他任意一个节点或者是为null。

 

（又是链表啊啊啊啊啊啊啊！！！！！！！！Orz……）

 

解题思路：

在每个原来节点的后面插入一个新节点（label值一样），然后复制原来节点的random指针，最后包含新旧链表的长链表分成一个原来的链表和一个新的链表。

参考这里面的两个图比较容易理解~……

有一点需要注意：就是遇到链表的问题时，一定要考虑p是空指针时，不要出现p->next之类的调用，本题也是一样要考虑的。

 

代码如下：

class Solution {
public:
    RandomListNode *copyRandomList(RandomListNode *head) {
        if(head == NULL)
            return NULL;
        RandomListNode *p = head;
        //第一遍：扫描顺序复制next指针
        while(p){
            RandomListNode *newNode = new RandomListNode(p->label);
            newNode->next = p->next;
            p->next = newNode;
            p = newNode->next;
        }
        p = head;
        //第二遍：复制random指针
        while(p){
            if(p->random)
                p->next->random = p->random->next;
            p = p->next->next;
        }
        p = head;
        //第三遍：恢复旧链表和新链表
        RandomListNode *newHead = p->next;
        RandomListNode *newP = newHead;
        
        p->next = newP->next;
        p = p->next;
        //要保证p不是空指针，不然调用p->next就会报错。
        while(p){
            newP->next = p->next;
            newP = newP->next;
            p->next = newP->next;
            p = p->next;
        }
        return newHead;
    }
};