【剑指Offer】复杂链表的复制

题目描述

输入一个复杂链表（每个节点中有节点值，以及两个指针，一个指向下一个节点，另一个特殊指针random指向一个随机节点），请对此链表进行深拷贝，并返回拷贝后的头结点。（注意，输出结果中请不要返回参数中的节点引用，否则判题程序会直接返回空）

解题思路

参考博主的思路，如图片的水印。

代码

/*
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 null;
        RandomListNode p = pHead;
        while(p != null) {
            RandomListNode cloneNode = new RandomListNode(p.label);
            RandomListNode nextNode = p.next;
            p.next = cloneNode;
            cloneNode.next = nextNode;
            p = nextNode;
        }

        p = pHead;
        while(p != null) {
            if(p.random == null) {
                p.next.random = null;
            } else {
                p.next.random = p.random.next;
            }
            p = p.next.next;
        }

        p = pHead;
        RandomListNode cHead = pHead.next;
        while(p != null) {
            RandomListNode cloneNode = p.next;
            p.next = cloneNode.next;
            cloneNode.next = cloneNode.next==null?null:cloneNode.next.next;
            p = p.next;
        }

        return cHead;
    }
}