牛客刷题之复杂链表的复制

题目描述

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

解题思路：

       首先复制原始链表的任意节点N并创建新节点N‘，在把N’链接到N的后面；

       接下来复制特殊节点，很容易看出，假设原始链表上的节点N 的特殊指针指向S，则它对应的复制节点N‘的特殊指针指向S的下一个节点S‘；

        最后进行原链表和新链表的分离，并将新链表的头节点进行返回。

/*
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 rln = pHead;

        //节点的复制
        while (rln != null) {
            RandomListNode re_node = rln.next;
            RandomListNode clone_node = new RandomListNode(rln.label);
            clone_node.next = re_node;
            rln.next = clone_node;
            rln = re_node;
        }
        //从头复制其特殊指针
        rln = pHead;
        while (rln != null) {
            if (rln.random != null) {
                rln.next.random = rln.random.next;
            }
            rln = rln.next.next;
        }
        //链表的分离
        rln = pHead;
        RandomListNode rlne = rln.next;
        RandomListNode clonehead = rlne;
        while (rln != null) {
            rln.next = rlne.next;
            rln = rln.next;
            if (rln == null) {
                rlne.next = null;
            } else {
                rlne.next = rln.next;
            }
            rlne = rlne.next;
        }
        return clonehead;
    }
}