剑指offer全集详解python版——复杂链表的复制

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

思路：

技巧性很强的以一个题：

• 复制原始链表的每个结点, 将复制的结点链接在其原始结点的后面
• 将复制后的链表中的复制结点的random指针链接到被复制结点random指针的后一个结点
• 拆分链表, 将原始链表的结点组成新的链表, 复制结点组成复制后的链表

代码：

# -*- coding:utf-8 -*-
# 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
        pNode = pHead
        if pHead == None:
            return None
        self.connectnodes(pHead)
        self.connectrandomnodes(pHead)
        return self.reconnectnodes(pHead)
    # 复制原始链表的每个结点, 将复制的结点链接在其原始结点的后面
    def connectnodes(self,pHead):
        pNode = pHead
        while (pNode):
            pClone = RandomListNode(0)
            pClone.label = pNode.label
            pClone.next = pNode.next
            pClone.random = None
            pNode.next = pClone
            pNode = pClone.next
# 将复制后的链表中的复制结点的random指针链接到被复制结点random指针的后一个结点
    def connectrandomnodes(self,pHead):
        pNode = pHead
        while (pNode):
            pClone = pNode.next
            if pNode.random:
                pClone.random = pNode.random.next
            pNode = pClone.next
# 拆分链表, 将原始链表的结点组成新的链表, 复制结点组成复制后的链表
    def reconnectnodes(self,pHead):
        pNode = pHead
        pHeadclone = pClone = pNode.next
        pNode.next = pClone.next
        pNode = pClone.next
        while pNode:
            pClone.next = pNode.next
            pClone = pClone.next
            pNode.next = pClone.next
            pNode = pNode.next
        return pHeadclone