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

题目链接

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

题解

题解1

一次遍历，通过修改链表结构，让每一个节点的复制节点先链到自己的next上
一次遍历，修改每一个复制节点的rand指针
一次遍历，修改每一个复制节点的next指针，并且还原原来链表

class Solution {
    public Node copyRandomList(Node head) {
        
        if (head == null) {
            return null;
        }

        // 给每一个节点都整一个copy节点
        Node cur = head;
        while (cur != null) {
            // 取出下一个节点
            Node next = cur.next;
            // 复制一个节点
            Node newNode = new Node(cur.val);
            // 新节点的next指针指向原来节点的next指针
            newNode.next = cur.next;
            // 本节点的next指向新节点
            cur.next = newNode;
            // 换头
            cur = next;
        }

        // 设置拷贝链表各个节点的random指针
        cur = head;
        while (cur != null) {
            // 拷贝节点
            Node copyNode = cur.next;
            // 设置random指针
            copyNode.random = cur.random == null ? null : cur.random.next;
            // 换头
            cur = copyNode.next;
        }

        cur = head;
        // 还原原链表以及修改复制链表的next指针
        Node resultHead = head.next;
        while (cur != null) {
            Node copyNode = cur.next;
            Node next = copyNode.next;
            copyNode.next = copyNode.next == null ? null : copyNode.next.next;
            cur.next = next;
            cur = next;
        }

        return resultHead;
    }
}

题解2

一次遍历，给每个节点创建一个复制节点，形成一一对应关系放到map中
一次遍历，给每一个复制节点设置next和rand

class Solution {
    public Node copyRandomList(Node head) {
        if (head == null)
            return head;

        // 创建一个hash表来存储链表和链表的拷贝节点
        Map<Node, Node> nodeAndCopyNode = new HashMap<>();

        // hash表初始化，遍历链表，先把每个链表的拷贝节点都创建出来
        Node cur = head;
        while (cur != null) {
            Node copyNode = new Node(cur.val);
            nodeAndCopyNode.put(cur, copyNode);
            cur = cur.next;
        }

        // 再次遍历链表，把拷贝节点的rand指针和next指针赋值
        cur = head;
        while (cur != null) {
            Node copyNode = nodeAndCopyNode.get(cur);
            copyNode.next = nodeAndCopyNode.get(cur.next);
            copyNode.random = nodeAndCopyNode.get(cur.random);
            cur = cur.next;
        }

        return nodeAndCopyNode.get(head);
    }
}