Leetcode 138. 复制带随机指针的链表 C++

Leetcode 138. 复制带随机指针的链表

题目

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

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

我们用一个由 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]]

示例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 。

题解

用一个哈希表实现对应关系，然后依据原先的链表把新节点连接起来就可。方法比较赖。。。详细过程见代码

代码

Node* copyRandomList(Node* head) {
        if(head == NULL)    return nullptr;
        map<Node*,Node*> m;
        Node* p=head;
        Node* newP;
        while(p){
            newP = new Node(p->val);
            m[p] = newP;
            p = p->next;
        }
        p = head;
        while(p){
            if(p->next) m[p]->next = m[p->next];
            else m[p]->next = NULL;
            if(p->random)   m[p]->random = m[p->random];
            else m[p]->random = NULL;
            p = p->next;
        }
        return m.find(head)->second;
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/copy-list-with-random-pointer
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。