剑指offer-25题-复杂链表复制

Posted 2021-03-05 forest-river

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

 

 

python代码：

# -*- coding:utf-8 -*-
# class RandomListNode:
#     def __init__(self, x):
#         self.label = x
#         self.next = None
#         self.random = None
class Solution:
    # 返回 RandomListNode
    # 返回 RandomListNode
    def newpath(self, p):
        while(p):
            new = RandomListNode(p.label)
            new.next = p.next
            p.next = new
            p = new.next

    def cloneRandom(self, p):
        while(p):
            if(p.random != None):
                p.next.random = p.random.next
            p = p.next.next

    def split(self, p):
        newhead = p.next
        p.next = newhead.next
        p = newhead
        while(p.next):
            ne = p.next
            p.next = ne.next
            p = ne
        return newhead

    def Clone(self, pHead):
        # write code here
        if(pHead == None):
            return None
        self.newpath(pHead)
        self.cloneRandom(pHead)
        newhead = self.split(pHead)
        return newhead