leetcode 探索 鏈表 複製帶隨機指針的鏈表

題目

給定一個鏈表，每個節點包含一個額外增加的隨機指針，該指針可以指向鏈表中的任何節點或空節點。
要求返回這個鏈表的 深拷貝。
我們用一個由 n 個節點組成的鏈表來表示輸入/輸出中的鏈表。每個節點用一個 [val, random_index] 表示：
val：一個表示 Node.val 的整數。
random_index：隨機指針指向的節點索引（範圍從 0 到 n-1）；如果不指向任何節點，則爲 null 。
示例1：

輸入：head = [[7,null],[13,0],[11,4],[10,2],[1,0]]
輸出：[[7,null],[13,0],[11,4],[10,2],[1,0]]

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/copy-list-with-random-pointer

分析

建立一個輔助空間map，遍歷舊的鏈表，然後依次存儲舊的節點與新的節點的映射關係。
爲啥要存儲呢？因爲我們後邊再遍歷新的節點時候，我們會取出它當前的Random，這個random還是舊的節點，我們通過map將這個舊的random節點替換爲新的節點。

注意：這裏不需要擔心設置的新Random節點，它的Random沒更新。因爲存儲的是指針引用，隨着新節點的遍歷，我們會修改掉每一個新的節點Random的。

總體上，空間複雜度O(N), 時間複雜度O(2N)

解法

/**
 * Definition for a Node.
 * type Node struct {
 *     Val int
 *     Next *Node
 *     Random *Node
 * }
 */

func copyRandomList(head *Node) *Node {
    if head == nil {
        return nil
    }
    // 如果遇到random，沒有的，先存起來，存到還需要設置的，有的就從已經設置的map裏取出，設置上去，
    var m = make(map[*Node]*Node)
    // var toM = make(map[*Node]*Node)
    cur := head
    var newHead = &Node{}
    var pre = newHead
    for cur != nil {
        mCur := &Node{
            Val: cur.Val,
            Random: cur.Random,
        }
        m[cur] = mCur
        pre.Next = mCur
        pre = pre.Next
        cur = cur.Next
    }
    
    newCur := newHead.Next
    for newCur != nil {
        if newCur.Random != nil {
            if newRandom, ok := m[newCur.Random]; ok {
                newCur.Random = newRandom
            }
        }
        newCur = newCur.Next
    }
    
    return newHead.Next
}