【Copy List with Random Pointer】cpp

题目：

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.

代码：

/**
 * 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) {
            if ( !head ) return head;
            std::map<RandomListNode *, RandomListNode *> oriNext;
            RandomListNode *pOri = head;
            RandomListNode dummy(-1);
            RandomListNode *pCopy = &dummy;
            // first round visit
            while (pOri)
            {
                // store the original Node's next pointer
                oriNext[pOri] = pOri->next;
                // create copy Node
                pOri->next = new RandomListNode(pOri->label);
                pCopy->next = pOri->next;
                pCopy->next->random = pOri;
                pCopy = pCopy->next;
                // move to the next Node
                pOri = oriNext[pOri];
            }
            // second round visit
            pCopy = dummy.next;
            while (pCopy)
            {
                pCopy->random = pCopy->random->random ? pCopy->random->random->next : NULL;
                pCopy = pCopy->next;
            }
            // third round recover original list
            for (std::map<RandomListNode *, RandomListNode *>::iterator i = oriNext.begin(); i != oriNext.end(); ++i)
            {
                i->first->next = i->second;
            }
            return dummy.next;
    }
};

 

Tips：

典型的链表深拷贝，之前用python做的时候用了O(1)空间技巧（http://www.cnblogs.com/xbf9xbf/p/4216655.html）。

这次尝试了用hashmap的O(n)空间的技巧，原因是感觉hashmap可能更通用一些。

思路主要参考了下面这个日志：http://www.geeksforgeeks.org/a-linked-list-with-next-and-arbit-pointer/。

需要注意的是：如果原来的链表节点random为空，就不要再往下寻找next；需要加一个保护判断

===============================================================

第二次过这道题，hashmap的大体思路画画图还记得，代码调试了几次才AC。

（1）不要忘记恢复原来的链表

（2）在拷贝random的时候，不要忘记copy = copy->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) {
            if (!head) return NULL;
            map<RandomListNode*, RandomListNode*> ori_next;
            RandomListNode* p = head;
            RandomListNode dummpy(-1);
            RandomListNode* copy = &dummpy;
            while (p)
            {
                copy->next = new RandomListNode(p->label);
                copy->next->random = p;
                ori_next[p] = p->next;
                p = p->next;
                copy->next->random->next = copy->next;
                copy = copy->next;
            }
            copy = dummpy.next;
            while (copy)
            {
                copy->random = copy->random->random ? copy->random->random->next : NULL;
                copy = copy->next;
            }
            for ( map<RandomListNode*, RandomListNode*>::iterator i=ori_next.begin(); i!=ori_next.end(); ++i )
            {
                i->first->next = ori_next[i->first];
            }
            return dummpy.next;
    }
};