Time: 3 ms (56.79%), Space: 5.4 MB (18.75%) - LeetHub

hovanhoa · Dec 27, 2024 · 4c474d0 · 4c474d0
1 parent 6c37136
commit 4c474d0
Showing 1 changed file with 9 additions and 10 deletions.
diff --git a/0138-copy-list-with-random-pointer/0138-copy-list-with-random-pointer.go b/0138-copy-list-with-random-pointer/0138-copy-list-with-random-pointer.go
@@ -8,22 +8,21 @@
  */
 
 func copyRandomList(head *Node) *Node {
-    oldToCopy := map[*Node]*Node{}
-    cur := head
+    oldToNew := map[*Node]*Node{}
 
+    cur := head
     for cur != nil {
-        copy := &Node{Val: cur.Val}
-        oldToCopy[cur] = copy
+        new := &Node{Val:cur.Val}
+        oldToNew[cur] = new
         cur = cur.Next
-    }
+    }   
 
-    cur = head
+    cur = head 
     for cur != nil {
-        copy := oldToCopy[cur]
-        copy.Next = oldToCopy[cur.Next]
-        copy.Random = oldToCopy[cur.Random]
+        oldToNew[cur].Next = oldToNew[cur.Next]
+        oldToNew[cur].Random = oldToNew[cur.Random]
         cur = cur.Next
     }
 
-    return oldToCopy[head]
+    return oldToNew[head]
 }