剑指 Offer 35. 复杂链表的复制（138. 复制带随机指针的链表）

剑指 Offer 35. 复杂链表的复制

请实现 copyRandomList 函数，复制一个复杂链表。在复杂链表中，每个节点除了有一个 next 指针指向下一个节点，还有一个 random 指针指向链表中的任意节点或者 null。

示例 1：

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

示例 2：

输入：head = [[1,1],[2,1]]
输出：[[1,1],[2,1]]

示例 3：

输入：head = [[3,null],[3,0],[3,null]]
输出：[[3,null],[3,0],[3,null]]

示例 4：

输入：head = []
输出：[]
解释：给定的链表为空（空指针），因此返回 null。

提示：

• -10000 <= Node.val <= 10000
• Node.random 为空（null）或指向链表中的节点。
• 节点数目不超过 1000 。

注意：本题与主站 138 题相同：力扣

思路：

• 先初始化map中的所有Node节点（这也是逐一复制不同于源节点的过程，指针需要重新创建）
• 连接各个单节点的Next和Random指针；因为一轮循环中已初始化，所以此时可以直接用

时间复杂度：两轮遍历链表，使用 O(N) 时间。

空间复杂度： 哈希表 nodeMap 使用线性大小的额外空间。

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

func copyRandomList(head *Node) *Node {
    nodeMap := make(map[*Node]*Node)

    // 先初始化map中的所有Node节点（这也是逐一复制不同于源节点的过程，指针需要重新创建）
    for cur := head; cur != nil; cur = cur.Next {
        nodeMap[cur] = &Node{}
    }

    for cur := head; cur != nil; cur = cur.Next {
        // 连接各个单节点的Next和Random指针；因为上面已初始化，所以此时可以直接用nodeMap[cur.Next]、nodeMap[cur.Random]...
        nodeMap[cur].Val, nodeMap[cur].Next, nodeMap[cur].Random = cur.Val, nodeMap[cur.Next], nodeMap[cur.Random]
    }

    return nodeMap[head]
}