剑指Offer:复杂链表的复制(35)

题目描述:

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

解答思路:

遍历原来的链表并拷贝每一个节点,将拷贝的节点放到之前节点的后面,新的链表是新旧节点交替的(A->A'->B->B'->C->C');

像这样:

然后迭代这个链表,让新结点跟旧结点随机指针指向相同

就像下面这个图一样:

最后从第一个开始解除链表以返回原始链表和克隆链表。

最后得到这样:

public class Solution {
  public Node copyRandomList(Node head) {

    if (head == null) {
      return null;
    }

    Node ptr = head;
    while (ptr != null) {

      // 克隆结点
      Node newNode = new Node(ptr.val);

      // 将克隆的结点放到原结点旁边
      newNode.next = ptr.next;
      ptr.next = newNode;
      ptr = newNode.next;
    }

    ptr = head;

    //迭代让新结点的随机指针指向跟旧结点相同
    while (ptr != null) {
      ptr.next.random = (ptr.random != null) ? ptr.random.next : null;
      ptr = ptr.next.next;
    }

    //解除链表，返回原始链表和克隆的链表。
    Node ptr_old_list = head; // A->B->C
    Node ptr_new_list = head.next; // A'->B'->C'
    Node head_old = head.next;
    while (ptr_old_list != null) {
      ptr_old_list.next = ptr_old_list.next.next;
      ptr_new_list.next = (ptr_new_list.next != null) ? ptr_new_list.next.next : null;
      ptr_old_list = ptr_old_list.next;
      ptr_new_list = ptr_new_list.next;
    }
    return head_old;
  }
}