力扣138-复制带随机指针的链表-C++

一、题目

题目较长，有需要的可以移步力扣官网：

力扣

二、解题思路

题目要求要对原链表进行深拷贝，所以新链表的指针不能指向原链表的结点，对应于原链表的每个结点都要在新链表中创建新的结点。

这里需要借助哈希表，以原链表中的结点为key，以新链表中对应的结点为value，用来存储原链表与新链表对应的结点对，（哈希表的主要作用是用于设置新链表的指针时进行查询），分两趟遍历原链表，第一趟每遍原链表的一个结点，就创建一个新节点，将原，新结点对插入到哈希表中；第二趟遍历原链表的每个结点时，若原链表结点的next或random指针不为空，则在哈希表中查到这两个指针对应的结点，然后指针指向的结点为索引，找到对应的value结点（该结点就为新链表中结点的next或random指针指向的结点）。

三、代码

class Solution {
public:
    Node* copyRandomList(Node* head) {
        // 借助哈希表
        if(head == NULL) return head;
        unordered_map<Node*, Node*> mp;  //存储每个原结点对应的新结点
        Node* p = head;  //遍历原链表
        while(p != NULL){  //将源结点与对应的新结点对放入到哈希表中
            Node* tmp = new Node(p->val);
            mp.insert(make_pair(p, tmp));
            p = p->next;
        }
        p = head;
        while(p != NULL){  //再次遍历原链表，设置新表的next和random
            if(p->next != NULL){
                mp[p]->next = mp[p->next];  
            }
            if(p->random != NULL){
                mp[p]->random = mp[p->random];
            }
            p= p->next;
        }
        return mp[head];
    }
};

四、运行结果