剑指offer【python】——复杂链表的复制

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

方法一

# -*- 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
        #链表为空
        if pHead == None:
            return None
        #ptemp指向头指针
        ptemp = pHead
        
        #将每个节点进行复制，并插入原链表
        while ptemp:
            #复制当前节点
            pcopy = RandomListNode(ptemp.label)
            #新节点插入原链表中(链表插入元素操作)
            pcopy.next = ptemp.next
            ptemp.next = pcopy
            #插入新节点，更新ptemp，此时ptemp跳过新加入的节点 ptemp = ptemp.next.next(等价于ptemp = pcopy.next)
            #ptemp.next:新加入的节点
            ptemp = pcopy.next
            
        #实现复制节点的random
        ptemp = pHead
        while ptemp:
            #先判断当前节点的random是否存在，存在则进行操作
            if ptemp.random:
                #ptem.next：复制节点
                #ptemp.next.random：复制节点的random
                #A->A'->B->B':若A的random指针指向B，那么A'则指向B'，此时只需要找到B'即可
                #因此ptemp.random指向B,ptemp.random.next为B'
                ptemp.next.random = ptemp.random.next
            #更新ptemp(A),ptemp.next.next则指向B’
            ptemp = ptemp.next.next
            
        #断开复制节点和原链表节点之间的连接
        ptemp = pHead
        #当A的指针指向B，A'需要新的头指针。ptemp2为新的头指针，ptemp2一直移动，最后的位置就变为尾指针
        ptemp2 = ptemp.next
        #最后需要返回头指针
        newHead = pHead.next
        while ptemp:
            #A->B
            ptemp.next = ptemp.next.next
            #当A'的指针指向B'时，先要判断B是否存在，若B不存在，那么B’肯定不存在
            if ptemp2.next:
                #断开A'->B,形成A'->B'
                ptemp2.next = ptemp2.next.next
                #对ptemp2更新，因为已经有A'->B',所以ptem2直接后移一位到B'
                ptemp2 = ptemp2.next
            #此时A->A'已经断开，形成A->B因此ptemp后移一位，为B
            ptemp = ptemp.next
        return newHead
        
        

方法二

该方法能通过，具体原理可看deepcopy源码

# -*- coding:utf-8 -*-
# class RandomListNode:
#     def __init__(self, x):
#         self.label = x
#         self.next = None
#         self.random = None
import copy
class Solution:
    # 返回 RandomListNode
    def Clone(self, pHead):
        # write code here
        result = copy.deepcopy(pHead)
        return result