【剑指offer刷题笔记】35.复杂链表的复制

剑指No.35_复杂链表的复制

• 题目：请实现 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]]

• 使用Map保存节点的方法

    public Node copyRandomListWay(Node head){
        if (head == null) return null;
        Map<Node, Node> map = new HashMap<>();
        Node temp = head;

// 将老节点与新复制的新节点存入map中
        while (temp != null){
            map.put(temp, new Node(temp.val));
            temp = temp.next;
        }

// 在map中把老节点的next和random复制给新节点
        temp = head;
        while (temp != null){
            map.get(temp).next = map.get(temp.next);
            map.get(temp).random = map.get(temp.random);
            temp = temp.next;
        }

        return map.get(head);
    }

• 交叉拼接+复制的方法，可以避免生成多余的map空间：

    public Node copyRandomListWay(Node head){
        if (head == null) return null;
        Node temp = head;

        // 第一步：把复制出来的新节点交叉的拼接在老节点之间
        while (temp != null){
            Node node = new Node(temp.val);
            node.next = temp.next;
            temp.next = node;
            temp = node.next;
        }

        temp = head;
        // 第二步：把老节点的random指针以同样的顺序复制到新节点的random指针上
        while (temp != null){
            if (temp.random != null)
                temp.next.random = temp.random.next;
            temp = temp.next.next;
        }

        temp = head.next;
        Node pre = head, res = head.next;
        // 第三步：把新老节点拆分成两个链表
        while (temp.next != null){
            pre.next = pre.next.next;
            temp.next = temp.next.next;
            pre = pre.next;
            temp = temp.next;
        }

        pre.next = null;
        return res;
    }