题目描述
输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
第一种方法,在原链表基础上复制后插入,再一次遍历保证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):
# write code here
if not pHead:
return None
temp=pre=n=pHead
while temp:
n=temp.next
node = RandomListNode(temp.label)#新建结点
node.next=n#新结点的next指向元结点的next
temp.next=node#原结点的next指向新结点
temp=n#向后移
pre=pHead
while pre:
temp=pre.next#复制结点是原结点next指向
prer=pre.random
if prer!=None:#random指针不为None说明是原结点
temp.random=prer.next#random指针指向原结点random的next指向
pre=temp.next#pre指向插入结点后一个,即原结点的后一个原结点
pre=pHead
head=pHead.next#指向复制链表的头结点
while pre:
temp=pre.next#指向原结点的下一个结点即复制结点
n=temp.next#指向下一个原结点
pre.next=n#从原链表中删除复制结点
if n:
temp.next=n.next#n不为None时说明后面还有结点未删除,指向下一个复制结点
else:
temp.next=None#n为空则已经删到最后,next指空
pre=n#指向下一个原结点
return head
第二种方法,利用额外空间制作哈希表,对应原结点和复制结点关系,生成链表时查找哈希表(本来想学一下这个方法,结果改到和别人的相似度99%了也过不了牛客的oj…先放在这改天再说
找了一下午终于找到问题在哪了,最开始初始化的时候要分开初始化,否则会导致都指向同一个内存空间
# -*- 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
nodelist=[]
randoml=[]
nodeval=[]#node保存结点,randoml保存其指向的random结点,nodeval保存结点值
head = pHead
while head:#遍历一次依次存入对应list
randoml.append(head.random)
nodelist.append(head)
nodeval.append(head.label)
head=head.next
randomIndex = map(lambda c: nodelist.index(c) if c else -1, randoml)#random指向结点的索引
nodelist=[]
for i in range(len(nodeval)):
node=RandomListNode(nodeval[i])
nodelist.append(node)
for i in range(len(nodelist)):
if randomIndex[i]!=-1:
nodelist[i].random=nodelist[randomIndex[i]]
head=nodelist[0]
for node in nodelist[1:]:
head.next=node
head=head.next
return nodelist[0]