版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/u014568072/article/details/87712688
题目描述
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
示例1
思路
《剑指offer》P147
参考书中的解析,一共分为三步:
- 对原表的每一个结点进行复制,并插入到被复制结点后。
- 根据原表中的每一个
random
结点,给复制结点赋予random
结点。 - 将复制的表与原表分离。
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