面试题35：复杂链表的复制

/*

 * 面试题35：复杂链表的复制

 * 题目：输入一个复杂链表（每个节点中有节点值，以及两个指针，一个指向下一个节点，另一个特殊指针指向任意一个节点），

 * 返回结果为复制后复杂链表的head。（注意，输出结果中请不要返回参数中的节点引用，否则判题程序会直接返回空）

 * 思路：在不用辅助空间的情况下，实现O(n)的时间效率。

 */

class RandomListNode {

    int label;

    RandomListNode next = null;

    RandomListNode random = null;

 

    RandomListNode(int label) {

       this.label = label;

    }

}

 

public class No35RandomListNode {

 

    public static void main(String[] args) {

       RandomListNode head = new RandomListNode(1);

       RandomListNode node2 = new RandomListNode(2);

       RandomListNode node3 = new RandomListNode(3);

       RandomListNode node4 = new RandomListNode(4);

       RandomListNode node5 = new RandomListNode(5);

      

       head.next = node2;

       node2.next = node3;

       node3.next = node4;

       node4.next = node5;

      

       head.random = node3;

       node2.random = node5;

       node4.random = node2;

      

       No35RandomListNode n = new No35RandomListNode();

      

       RandomListNode copyList = n.Clone(head);

       

       while (copyList != null) {

           System.out.print(copyList.label + " ");

 

           copyList = copyList.next;

       }

       

    }

 

    public RandomListNode Clone(RandomListNode pHead) {

       CloneNodes(pHead);

       ConnerSiblingNodes(pHead);

       return ReconnectNodes(pHead);

    }

   

    // 第一步：根据原始链表的每个节点N创建对应的节点N'

    private void CloneNodes(RandomListNode pHead) {

       RandomListNode pNode = pHead;

       while (pNode != null) {

           RandomListNode pCloned = new RandomListNode(pNode.label);

           pCloned.label = pNode.label;

           pCloned.next = pNode.next;

           pCloned.random = null;

           //把N'链接在N的后面

           pNode.next = pCloned;

           pNode = pCloned.next;

          

       }

      

    }

    // 第二步：设置复制出来节点的random.假设原始节点的随机指向S，复制出来节点的random指向S后一个

    private void ConnerSiblingNodes(RandomListNode pHead) {

       RandomListNode pNode = pHead;

       while (pNode != null) {

           RandomListNode pCloned = pNode.next;

           if (pNode.random != null) {

              pCloned.random = pNode.random.next;

           }

          

           pNode = pCloned.next;

       }

      

    }

 

 

    // 第三步：把这个长链表拆分成两个链表，奇数位置连接起来就是原始链表，偶数结点连接起来就是复制结点

    private RandomListNode ReconnectNodes(RandomListNode pHead) {

       RandomListNode pNode = pHead;//原节点

       RandomListNode pClonedHead = null;//要返回的节点

       RandomListNode pClonedNode = null;//借助的空闲节点

      

       if (pNode != null) {

           pClonedHead = pClonedNode = pNode.next;

           pNode.next = pClonedNode.next;

           pNode = pNode.next;

       }

       //该while与上一个if分开的原因是题目要求返回复杂链表的head，而非引用

       while (pNode != null) {

           pClonedNode.next = pNode.next;

           pClonedNode = pClonedNode.next;

           pNode.next = pClonedNode.next;

           pNode = pNode.next;

       }

       return pClonedHead;

    }

 

}