剑指offer面试题35 复杂链表的复制

题目描述：

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

解题思路：

1 哈希表

第一步复制原始链表上的每个节点N，由此创建复制节点N’，同时将<N，N’>对放到一个哈希表中。第二步利用已生成的哈希表对复制节点N’进行连接。利用空间换取时间，时间复杂度与空间复杂度都为O(n)。

代码

class Solution {
public:
    Node* copyRandomList(Node* head) {
        if(head == NULL) return NULL;
        unordered_map<Node*, Node*> newHead;
        Node* cur = head;
        while(cur){ //创建新节点
            newHead[cur] = new Node(cur->val);
            cur = cur->next;
        }
        cur = head;
        while(cur){ //进行next与random的连接
            newHead[cur]->next = newHead[cur->next];
            newHead[cur]->random = newHead[cur->random];
            cur = cur->next;
        }
        return newHead[head];
    }
};