剑指offer第二版（Python3）--面试题35：复杂链表的复制

第2章 面试需要的基础知识

第3章 高质量的代码

第4章 解决面试题的思路

  面试题27:二叉树的镜像

  面试题29:顺时针打印矩阵

  面试题30 :包含min函数的栈

  面试题31:栈的压入、弹出序列

  面试题32:上往下打印二叉树

  面试题33: 二叉搜索树的后序遍历序列

  面试题34: 二叉树中和为某一值的路径

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

第5章 优化时间和空间效率

第6章 面试中的各项能力

第7章 两个面试案例

---

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

解题思路
《剑指offer》
  一个含有5个节点的复杂链表如下：

  最自然的想法是分成两步进行：第一步复制链表上每个节点，并用p_next连接起来，第二步是复制每个节点p_random指针。由于p_random指针指向的节点可能在当前节点前面或者后面，所以要定位p_random指向的节点的位置需要从原始链表的头结点开始，因此时间复杂度为O(n²)。
  为了减少查找p_random指向节点的时间，可以将新旧节点作为一对存储到一个哈希表中，然后使用O(1)时间就可以找到p_random节点。
  在不使用辅助空间的前提下，我们还是可以做到O(n)的时间复杂度。我们每次把新复制的节点链接在原节点后面，如下所示：

  然后再复制p_random信息。根据上图可以发现原始链表上A的p_random指向C节点，那么它对应复制出来的节点A’是C节点的下一个节点，根据这种关系就很容易找到p_random信息了。
  复制完链表后，还需要拆分链表。需要注意的是必须等复制完全部节点后才能拆分节点，要不然如果存在p_random指向前面节点的情况，复制就会出错。

实战

# class RandomListNode:
#     def __init__(self, x):
#         self.label = x
#         self.next = None
#         self.random = None
class Solution:
    # 返回 RandomListNode
    def Clone(self, pHead):
        # write code here
        if not pHead:
            return None
        
        # 复制链表,random指针不复制
        p = pHead
        while p:
            _nex = p.next
            node = RandomListNode(p.label)
            p.next = node
            node.next = _nex
            p = _nex
        
        # 复制random指针
        newHead = pHead.next
        p = pHead
        while p:
            newp = p.next
            if p.random:
                newp.random = p.random.next
            p = newp.next
            
        # 拆分链表
        p = pHead
        newp = newHead
        while p:
            p.next = newp.next
            p = newp.next
            if p:
                newp.next = p.next
                newp = p.next

        return newHead