题目描述
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针random指向一个随机节点),请对此链表进行深拷贝,并返回拷贝后的头结点。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
解题思路
1.在每个节点后面插入复制的结点
2.把复制的结点的random指针指向原来结点的random指针指向的结点的下一个结点
3.拆分为两个链表
# -*- 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):
# write code here
if not pHead:
return None
#1.在每个结点后面插入要复制的结点
cur = pHead
while cur:
temp = RandomListNode(cur.label)
temp.next = cur.next
cur.next = temp
cur = temp.next
#2.把复制的结点的random指针指向原节点random指针所指向结点的下一个结点
cur = pHead
while cur:
temp = cur.next
if cur.random:
temp.random = cur.random.next
cur = temp.next
#3.拆分为两个链表
cur = pHead
pCloneHead = cur.next
while cur:
temp = cur.next
curNext = temp.next
cur.next = curNext
if curNext:
temp.next = curNext.next
else:
temp.next = None
cur = curNext
return pCloneHead