LeetCode 138:复制带随机指针的链表

https://leetcode-cn.com/probl...

一、题目形容

给你一个长度为 n 的链表,每个节点蕴含一个额定减少的随机指针 random ,该指针能够指向链表中的任何节点或空节点。

结构这个链表的深拷贝。深拷贝应该正好由 n 个 全新节点组成,其中每个新节点的值都设为其对应的原节点的值。新节点的 next 指针和 random 指针也都应指向复制链表中的新节点,并使原链表和复制链表中的这些指针可能示意雷同的链表状态。复制链表中的指针都不应指向原链表中的节点。

例如,如果原链表中有 X 和 Y 两个节点,其中 X.random --> Y 。那么在复制链表中对应的两个节点 x 和 y ,同样有 x.random --> y 。

返回复制链表的头节点。

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

val:一个示意 Node.val 的整数。

random_index:随机指针指向的节点索引(范畴从 0 到 n-1);如果不指向任何节点,则为  null 。

你的代码 只 承受原链表的头节点 head 作为传入参数。

节点定义

class Node {    int val;    Node next;    Node random;    public Node(int val) {        this.val = val;        this.next = null;        this.random = null;    }}

二、示例

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

三、思路

额定空间复杂度为O(1)的写法

1、在原链表每个节点前面,复制一个新的节点,并连到老节点前面

即:

复制前:1 -> 2 -> 3,

复制后:1 -> 1\` -> 2 -> 2\` -> 3 -> 3\`

2、将原节点random指针的关系,在新节点中复制一份

3、拆散新老节点

额定空间复杂度为O(N)的写法

1、应用Map存储新老节点,Key:老节点,Value:新节点

2、结构老节点对应的新节点

3、调整新节点的next指针、调整新节点的random指针

四、源码

额定空间复杂度为O(1)的写法

/** * @author Java和算法学习:周一 */public Node copyRandomList(Node head) {    if (head == null) {        return null;    }    // 1.将新节点插入到老节点前面    // 调整前:1 -> 2 -> 3 -> null,调整后:1 -> 1` -> 2 -> 2` -> 3 -> 3` -> null    Node current = head;    Node next = null;    while (current != null) {        // 标记原链表中以后节点的next节点        next = current.next;        // 在原链表以后节点的前面插入新节点        current.next = new Node(current.val);        // 调整新插入节点的next节点        current.next.next = next;        // 以后节点后移        current = next;    }    // 2.调整新插入节点的random指针    Node copy = null;    current = head;    while (current != null) {        // 新节点        copy = current.next;        // 标记原链表中以后节点的next节点        next = current.next.next;        // 调整新节点的random节点        copy.random = current.random != null ? current.random.next : null;        // 以后节点后移        current = next;    }    // 3.拆散新老节点    current = head;    // 记录新链表的头节点    Node res = head.next;    while (current != null) {        // 新节点        copy = current.next;        // 标记原链表中以后节点的next节点        next = current.next.next;        // 以后节点的next节点指向原链表中以后节点的next节点(行将老节点的援用调整回去)        current.next = next;        // 新节点的next节点指向 原链表中以后节点next节点 的next节点(即抽出新节点的援用)        copy.next = next != null ? next.next : null;        // 以后节点后移        current = next;    }    // 返回新链表的头节点    return res;}

额定空间复杂度为O(N)的写法

/** * @author Java和算法学习:周一 */public Node copyRandomListByMap(Node head) {    // Key:老节点,Value:新节点    HashMap<Node, Node> map = new HashMap<>();    Node current = head;    while (current != null) {        // 结构老节点对应的新节点        map.put(current, new Node(current.val));        current = current.next;    }    current = head;    while (current != null) {        // 调整新节点的next指针        map.get(current).next = map.get(current.next);        // 调整新节点的random指针        map.get(current).random = map.get(current.random);        current = current.next;    }    return map.get(head);}

五、测试后果

额定空间复杂度为O(1)的写法

额定空间复杂度为O(N)的写法

从LeetCode官网的测试后果能够看出,执行用时没有差距,内存耗费(即额定空间复杂度)的差距就很显著了,一个战胜 99.94% 的用户,一个只战胜了 48.64% 的用户。