复杂链表的赋值

请实现copyRandomList函数,复制一个复杂链表。在复杂链表中,每个节点除了有一个next指针指向下一个节点,还有一个random指针指向链表中的任意节点或者null。
在这里插入图片描述

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

思路:哈希表

复制普通链表很简单,只需遍历链表,每轮建立新节点 + 构建前驱结点pre和当前节点node的引用指向即可。

我们用哈希表来解决这个问题:
首先创建一个哈希表,再遍历原链表,遍历的同时再不断创建新节点
我们将原节点作为key,新节点作为value放入哈希表中。
在这里插入图片描述
第二步我们再遍历原链表,这次我们要将新链表的next和random指针给设置上。
在这里插入图片描述
从上图中,可以发现,原节点和新节点是一一对应关系,所以,

  • map[cur->val],得到的就是对应的新节点
  • map[cur->next],得到的就是对应的新节点.next
  • map[cur->random],得到的就是对应的新节点.random
    所以,我们只需要再次遍历原链表,然后设置:
    新节点.next -> map[cur->next]
    新节点.random -> map[cur->random]
    最后,map[head],也就是对应的新链表的头结点返回。

利用哈希表的查询特点,考虑构建原链表节点新链表对应节点的键值对映射关系,再遍历构建新链表各节点的next和random引用指向即可。
算法流程:

  1. 若头节点head为空节点,直接返回null
  2. 初始化: 哈希表dic,节点cur指向头节点
  3. 复制链表:
  4. 构建新链表的引用指向:
  5. 返回值:新链表的头节点dic[cur];
/*
// Definition for a Node.
class Node {
public:
    int val;
    Node* next;
    Node* random;
    
    Node(int _val) {
        val = _val;
        next = NULL;
        random = NULL;
    }
};
*/
class Solution {
public:
    Node* copyRandomList(Node* head) {
        if(head == nullptr) return nullptr;
        Node* cur = head;
        unordered_map<Node*, Node*> map;
        // 3. 复制各节点,并建立 “原节点 -> 新节点” 的 Map 映射
        while(cur != nullptr) {
            map[cur] = new Node(cur->val);
            cur = cur->next;
        }
        cur = head;
        // 4. 构建新链表的 next 和 random 指向
        while(cur != nullptr) {
            map[cur]->next = map[cur->next];
            map[cur]->random = map[cur->random];
            cur = cur->next;
        }
        // 5. 返回新链表的头节点
        return map[head];
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值