剑指offer：Python 复杂链表的复制 图解 实现过程

题目描述

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

解读题意

首先 复杂链表，除了有一个指向下一个结点的指针外，还有一个特殊（任意）指针，它可以指向任意位置，如下图；其次时间复杂度如果为n^2的话，则不是最佳方法

思路及Python实现

拷贝结点 实现

• 将原链表所有的结点拷贝一份，并且每个拷贝的结点和原来的结点连续出现，如下图：
  

• 此时，原来结点指向下一个结点的指针都可以先链接上，原来的random指针也还在；但是每个复制新建的node的random区域还是空的

• 找出规律：A.next.random = a.random.next
  理解：如上图：A.next.raomd，是指第二个“1结点”的random指针，它应该指向 第一个“1结点”的random（第一个“3结点”）后一个结点（即第二个“3结点”），所以有上述变化通式

• 让每两个相同的结点断开，具体做法为：比如第一个“1结点” 让它指向 第一个“2结点”，random也跟着原来的指向变化，后面的结点重复操作，就能达到下图示意的，复制了复杂链表的操作

Python 实现过程

class RandomListNode:
    def __init__(self, x):
        self.label = x
        self.next = None
        self.random = None

class Solution:
    # 返回 RandomListNode
    def Clone(self, pHead):
        if pHead is None:
            return None
            
		# 为每一个结点复制一个node，并插入到原链表中
        pTemp = pHead 
        while pTemp:
            node = RandomListNode(pTemp.label)
            node.next = pTemp.next # 如下图2-1
            pTemp.next = node # 如下图 2-2
            pTemp = node.next # 如下图2-3：完成第一个新node插入，移动pTemp，重复上述，完成第二个
            
		# 实现新建node的random指向
        pTemp = pHead 
        while pTemp:
            if pTemp.random: # 如果random指针存在
                pTemp.next.random = pTemp.random.next # 图2-4：通过前一个random找到node的random
            pTemp = pTemp.next.next # 完成一个，移动pTemp帮助找到下一个的random
		
		# 断开 原来node 和新 node 间 的链接
        pTemp = pHead 
        newHead = pHead.next # 这个是新链表的头，到最后返回，一般要保留
        pNewTemp = pHead.next # 再新建一个中间变量来存储新的node
        while pTemp:
            pTemp.next = pTemp.next.next # 改变前一个node的指向，让它不要指向新建的node
            if pNewTemp.next: # 如果下个一的新node存在 （备注 1）
                pNewTemp.next = pNewTemp.next.next  # （备注 2）
                pNewTemp = pNewTemp.next # （备注 3）
            pTemp = pTemp.next # （备注 4） 备注（1~4）见下备注图解
        return newHead

# 测试一下
if __name__ == '__main__':
    n1 = RandomListNode(1)
    n2 = RandomListNode(2)
    n3 = RandomListNode(3)
    n4 = RandomListNode(4)
    n5 = RandomListNode(5)

    n1.next = n2
    n2.next = n3
    n3.next = n4
    n4.next = n5

    s = Solution()
    new_head = s.Clone(n1)
    res = new_head
    while res:
        print(res.label)
        res = res.next  # 输出结果 1 2 3 4 5 说明复制成功

图2-1：

图2-2：

图2-3

图2-4

备注1~4 图解（省略了结点分区（数据和链接区都画一块）、random可以指向任意，不做标明）

依次往后，断开链接，就完成了复制

递归实现

class RandomListNode:
    def __init__(self, x):
        self.label = x
        self.next = None
        self.random = None

class Solution:
    # 返回 RandomListNode
    def Clone(self, pHead):
        if pHead == None:
            return None
        new_head = RandomListNode(pHead.label)  # 新建一个链表的头结点
        new_head.random = pHead.random # 令头结点的random等于原来头节点的random
        # new_head.next = pHead.next  令头结点的next等于原来头结点的next;
        # 直接让new_head.next等于递归后的pHead.next)
        new_head.next = self.Clone(pHead.next)  # 递归处理new_head.next
        return new_head # 递归直接四行解决了，现在看下思路一，简直了。。。

# 测试一下
if __name__ == '__main__':
    n1 = RandomListNode(1)
    n2 = RandomListNode(2)
    n3 = RandomListNode(3)
    n4 = RandomListNode(4)
    n5 = RandomListNode(5)

    n1.next = n2
    n2.next = n3
    n3.next = n4
    n4.next = n5

    s = Solution()
    new_head = s.Clone(n1)
    res = new_head
    while res:
        print(res.label)
        res = res.next # 输出结果：1 2 3 4 5 复制成功