【剑指offer】26. 复杂链表的复制

题目描述

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

思路

《剑指offer》P147
参考书中的解析，一共分为三步：

1. 对原表的每一个结点进行复制，并插入到被复制结点后。
2. 根据原表中的每一个random结点，给复制结点赋予random结点。
3. 将复制的表与原表分离。

code

# -*- 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):
        if not pHead:
            return None
        # write code here
        self.CloneNodes(pHead)
        self.CloneRandomNode(pHead)
        return self.SplitList(pHead)
        
    def CloneNodes(self, pHead):
        # 复制每一个节点并将其接在该节点后面
        root = pHead
        while root:
            node = RandomListNode(root.label)
            node.next = root.next
            root.next = node
            root = node.next
    
    def CloneRandomNode(self, pHead):
        node = pHead
        while node:
            cloneNode = node.next
            randomNode = node.random
            if randomNode:
                cloneNode.random = randomNode.next
            node = cloneNode.next
        
    def SplitList(self, pHead):
        node = pHead
        res = node.next
        while node:
            cloneNode = node.next
            if cloneNode.next:
                node.next = cloneNode.next
                cloneNode.next = node.next.next
            else:
                node.next = None
            node = node.next
        return res