剑指offer—复制带随机指针的链表

剑指offer—复制带随机指针的链表

题目描述

给定一个链表，每个节点包含一个额外增加的随机指针，该指针可以指向链表中的任何节点或空节点。

要求返回这个链表的 深拷贝。

我们用一个由 n 个节点组成的链表来表示输入/输出中的链表。每个节点用一个 [val, random_index] 表示：

val：一个表示 Node.val 的整数。
random_index：随机指针指向的节点索引（范围从 0 到 n-1）；如果不指向任何节点，则为 null 。

思路

假如没有随机指针，那么我们直接一边遍历链表一边复制就可以了。但是题目有一个随机指针，还是按照原来的思路的话，假如随机指针指到的结点还未创建，那么就会乱了套，所以需要换一个思路。
利用一个hash表来存储原结点和拷贝结点，其中原结点作为key，拷贝结点作为value

然后再从头结点head遍历，ump[cur]->next = ump[cur->next]； ump[cur]->random = ump[cur->random]；cur = cur->next；

代码

	Node* copyRandomList(Node* head) {
        if(head == NULL) return head;
		//1. 创建hashMap
        unordered_map<Node* ,Node*> ump;
        Node* curr = head;
        //2. 遍历链表，将原节点作为key，拷贝节点作为value保存在map中
        while(curr != NULL)
        {
            Node* copy = new Node(curr->val);
            ump[curr] = copy;
            curr = curr->next;
        }
		//3. 复制链表next和random指针
        curr = head;
        while(curr != NULL)
        {
            ump[curr]->next = ump[curr->next];
            ump[curr]->random = ump[curr->random];
            curr = curr->next;
        }

        return ump[head];
    }

复杂度

时间复杂度 $O(N)$

空间复杂度 $O(N)$