[LeetCode] Copy List with Random Pointe

 1 /**
 2  * Definition for singly-linked list with a random pointer.
 3  * struct RandomListNode {
 4  *     int label;
 5  *     RandomListNode *next, *random;
 6  *     RandomListNode(int x) : label(x), next(NULL), random(NULL) {}
 7  * };
 8  */
 9  
10 class Solution {
11 public:
12     RandomListNode *copyRandomList(RandomListNode *head) {
13         
14         RandomListNode* pOld = head;
15         RandomListNode* pNew = NULL;
16         RandomListNode* pRes = NULL;
17         
18         if(head == NULL) return NULL;
19         
20         // insert every new node after old new node
21         while(pOld)
22         {
23             pNew = new RandomListNode(pOld->label);
24             if(pOld == head) pRes = pNew;
25             pNew->next = pOld->next;
26             pOld->next = pNew;
27             pOld = pNew->next;
28         }
29         
30 
31         pOld = head;
32         // constrct new‘s random pointer
33         while(pOld)
34         {
35             pNew = pOld->next;
36             if(pOld->random == NULL)
37                 pNew->random == NULL;
38             else 
39                 pNew->random = pOld->random->next;
40             pOld = pNew->next;
41         }
42 
43 
44         // recover old‘s and new‘s next pointer
45         //恢复old list 和new list
46   
47         pOld = head;
48         
49         while(pOld)
50         {
51             pNew = pOld->next;
52             if(pNew == NULL)
53                 pOld->next = NULL;
54             else
55                 pOld->next = pNew->next;
56             pOld = pNew;
57         }
58 
59         return pRes;   
60     }
61 };