modify copy-list-with-random-pointer.c

Author: lightmen

copy-list-with-random-pointer.c

@@ -8,14 +8,14 @@
  */
 struct RandomListNode *copyRandomList(struct RandomListNode *head) {
     struct RandomListNode root;
-    struct RandomListNode *prev,*tmp,*p,*tmp_head;
+    struct RandomListNode *prev,*tmp,*p;
     struct RandomListNode **arr;
     int count = 0;
     int i;
+
     if(!head)
         return NULL;
 
-    root.next = NULL;
     prev = &root;
     while(head){
         tmp = (struct RandomListNode *)malloc(sizeof(struct RandomListNode));
@@ -35,6 +35,7 @@ struct RandomListNode *copyRandomList(struct RandomListNode *head) {
     arr = (struct RandomListNode **)malloc(sizeof(struct RandomListNode *) * count);
     i = 0;
     head = root.next;
+
     while(head){
         p = head->random;
         tmp = p->random;
@@ -45,7 +46,10 @@ struct RandomListNode *copyRandomList(struct RandomListNode *head) {
 
     for(i = 0; i < count-1; ++i)
         arr[i]->next = arr[i+1];
+
     arr[count-1]->next = NULL;
     head = arr[0];
+    free(arr);
+
     return root.next;
 }