题目描述
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
思路:
关于链表的相关操作一定要注意的就是指针的使用,遇到复杂的问题我们可以将问题分成几个部分分别处理,然后拼接起来就可以解决复杂问题了。
第一步:在原链表的基础上,在每个结点后面复制与其相同的结点
第二步:因为上面的复制只复制了label和next值并没有复制random,所以接下来要设置复制结点的随机链接
第三步:将上面已经完整复制好的结点分离出来组成复制链表。也就是将偶数位上的结点拿出来再进行链接。
# -*- 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):
self.cloneNodes(pHead)
self.connectSiblingNodes(pHead)
#第一步,在每个结点后复制结点
def cloneNodes(self, pHead):
pNode = pHead #这样是为了避免改变pHead
while pNode:
pCloned = RandomListNode(pNode.label) #利用链表类生成对应值的结点
pCloned.next = pNode.next
pNode.next = pCloned
pNode = pCloned.next
#第二步,设置复制的结点的随机random
def connectSiblingNodes(self, pHead):
pNode = pHead #同样为了避免改变pHead
while pNode:
pclone = pNode.next #这样是为了避免后面忽略出错,也可以不设置这,只要你后面理清楚pNode的移动就好(最后一步)
if pNode.random:
pclone.random = pNode.random.next
pNode = pclone.next #注意pNode不是跳到pNode.next,而是直接跳到了pNode.next.next
#第三步,将复制结点分离出来进行链接组成复制链表
def reconnectNodes(self, pHead):
pNode = pHead
pCloneHead = None #定义两个指针其实就是,一个头指针(因为最后要输出头指针),一个是结点指针
pCloneNode = None
if pNode:
pCloneHead = pCloneNode = pNode.next #开始先让两个指针指向第一个复制结点
pNode.next = pCloneNode.next
pNode = pNode.next #避免重复一些操作,所以这样设置以后就可以接着扫描作用后面的结点了,因为第一个结点已经操作过了
while pNode:
pCloneNode.next, pCloneNode = pNode.next, pCloneNode.next
pNode.next, pNode = pCloneNode.next, pNode.next
return pCloneHead