复杂链表的复制

题目描述

输入一个复杂链表（每个节点中有节点值，以及两个指针，一个指向下一个节点，另一个特殊指针指向任意一个节点），返回结果为复制后复杂链表的head。（注意，输出结果中请不要返回参数中的节点引用，否则判题程序会直接返回空）

https://www.nowcoder.com/practice/f836b2c43afc4b35ad6adc41ec941dba

解题想法

比较有技巧，放一张图就知道怎么做了

分为 3 步：

1. 插入复制节点

2. 复制 random 指向

3. 将两个序列进行分割

代码实现

/*
public class RandomListNode {
    int label;
    RandomListNode next = null;
    RandomListNode random = null;

    RandomListNode(int label) {
        this.label = label;
    }
}
*/
public class Solution {
    public RandomListNode Clone(RandomListNode pHead)
    {
        if (pHead == null){
            return pHead;
        }
        // 1. 对每个节点进行复制，链接到原节点后面
        RandomListNode head1 = pHead;
        while (head1 != null){
            RandomListNode temp = new RandomListNode(head1.label);
            temp.next = head1.next;
            head1.next = temp;
            head1 = temp.next;
        }
        
        // 2. 复制random指针，
        RandomListNode head2 = pHead;    
        while (head2 != null){
            RandomListNode temp = head2.random;
            if (temp != null){
                head2.next.random = temp.next;
            }else{
                head2.next.random = null;
            }
            head2 = head2.next.next;
        }
        
        // 3. 将复制的节点进行拆分出来
        RandomListNode head3 = pHead;
        RandomListNode head4 = pHead.next;
        while (head3 != null){
            RandomListNode temp = head3.next;
            head3.next = temp.next;
            if (temp.next != null){
                temp.next = temp.next.next;
            }else{
                temp.next = null;
            }
            head3 = head3.next;
        }
        return head4;        
    }
}