题目:
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
思路一:
哈希表
代码:
# 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
clonehead = RandomListNode(pHead.label)
node = clonehead
randomlist = {}
randomlist[pHead] = node
while pHead.next:
pHead = pHead.next
node.next = RandomListNode(pHead.label)
node = node.next
randomlist[pHead] = node
for i in randomlist:
if i.random:
randomlist[i].random = randomlist[i.random]
return clonehead
思路二:
复制每个节点后将复制节点放在原节点后,最后再拆分
代码:
# 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
p = pHead
while p:
node = RandomListNode(p.label)
node.next = p.next
p.next = node
p = node.next
p = pHead
while p:
if p.random:
p.next.random = p.random.next
p = p.next.next
clonehead = pHead.next
node = pHead
while node.next:
temp = node.next
node.next = node.next.next
node = temp
return clonehead
思路三:
不要脸写法
代码:
# class RandomListNode:
# def __init__(self, x):
# self.label = x
# self.next = None
# self.random = None
import copy
class Solution:
# 返回 RandomListNode
def Clone(self, pHead):
clonehead = copy.deepcopy(pHead)
return clonehead