python解决复杂链表的复制

Posted 2020-10-29 clouds-114

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

思路：第一步在原链表的基础上复制节点，将节点复制在原节点的后面。第二步复制随机节点。 第三步将新旧链表分离。 图示如下

代码如下：

#encoding:utf8
class  RandomListNode():
    def __init__(self,x):
        self.label = x
        self.next = None
        self.random = None
class Solution:
    def clone(pHead):
        if pHead == None:
            return None
        #复制节点在原节点之后
        pCur = pHead
        while(pCur != None):
            node = RandomListNode(pCur.label)
            node.next = pCur.next
            pCur.next = node
            pCur = node.next
        #复制random节点
        pCur = pHead
        while(pCur != None):
            if pCur.random != None:
                pCur.next.random = pCur.random.next
            pCur = pCur.next.next
        head = pHead.next
        cur = head
        #将新旧链表分离
        pCur = pHead
        while(pCur != None):
            pCur.next = pCur.next.next
            if cur.next != None:
                cur.next = cur.next.next
            cur = cur.next
            pCur = pCur.next
        return head