【题目】
一种特殊的链表节点类型描述如下:
class Node:
def __init__(self, data):
self.val = data
self.next = None
self.rand = None
Node类中的value是节点值,next指针和正常单链表中next指针的意义一样,都指向下一个节点,rand指针是Node类中新增的指针,这个指针可能指向链表中的任意一个节点,也可能指向None。
给定一个由Node节点类型组成的无环单链表的头节点head,请实现一个函数完成这个链表中所有结构的复制,并返回复制的新链表的头节点。
进阶:不使用额外的数据结构,只用有限几个变量,且在时间复杂度为O(N)内完成原问题要实现的函数。
【基本思路】
原问题。使用一个哈希表,建立旧节点和新节点的对应关系。然后遍历一遍链表,根据原链表和哈希表设置每一个新节点的next和rand指针。
#python3.5
class RandNode:
def __init__(self, data):
self.val = data
self.next = None
self.rand = None
def copyListWithRand1(head):
if head == None:
return None
map = {}
cur = head
while cur != None:
map[cur] = RandNode(cur.val)
cur = cur.next
cur = head
while cur != None:
map[cur].next = None if cur.next == None else map[cur.next]
map[cur].rand = None if cur.rand == None else map[cur.rand]
cur = cur.next
return map[head]
进阶问题。在原链表中每一个节点的后面插入一个副本节点。遍历一遍链表,在遍历时设置每一个副本节点的rand指针。最后再分离成两个链表即可。
def copyListWithRand2(head):
if head == None:
return None
cur = head
while cur != None:
next = cur.next
cur.next = RandNode(cur.val)
cur.next.next = next
cur = next
cur = head
while cur != None:
cur.next.rand = None if cur.rand == None else cur.rand.next
cur = cur.next.next
copyHead = head.next
cur = head
while cur != None:
next = cur.next
cur.next = next.next
next.next = None if next.next == None else next.next.next
cur = cur.next
return copyHead