LeetCode 高级 - 复制带随机指针的链表

复制带随机指针的链表

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

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

分析

该题参考 剑指Offer 复杂链表的复制。

思路：

1. 在原链表中插入“影子节点”
2. 根据原节点random 更新影子节点的random
3. 分离原链表和“影子链表”，该链表即为深度拷贝的链表

图示：

代码

/**
 * Definition for singly-linked list with a random pointer.
 * class RandomListNode {
 *     int label;
 *     RandomListNode next, random;
 *     RandomListNode(int x) { this.label = x; }
 * };
 */
public class Solution {
    public RandomListNode copyRandomList(RandomListNode head) {
        if(head ==null){
            return null;
        }
        RandomListNode p = head;

        //1.新节点接到原对应节点的后面。
        while(p != null){
            RandomListNode clone = new RandomListNode(p.label);
            clone.next = p.next;
            //clone.random = null;

            p.next = clone;
            p = clone.next;
        }

        //2.参照原节点的random，改变新节点的rondom
        p = head;
        while(p!=null){
            p.next.random = p.random==null?null:p.random.next;
            p=p.next.next;
        }

        //3.将两部分分离
        p = head;
        RandomListNode cloneHead = head.next;
        RandomListNode cloneNode = cloneHead;
        while(p!=null){
            p.next = cloneNode.next;
            if(cloneNode.next!=null){
                cloneNode.next = cloneNode.next.next;
            }            
            p = p.next;
            cloneNode = cloneNode.next;
        }

        return cloneHead;
    }
}