题目链接:https://leetcode-cn.com/problems/fu-za-lian-biao-de-fu-zhi-lcof/
Description
请实现 copyRandomList 函数,复制一个复杂链表。在复杂链表中,每个节点除了有一个 next 指针指向下一个节点,还有一个 random 指针指向链表中的任意节点或者 null。
示例 1:
输入:head = [[7,null],[13,0],[11,4],[10,2],[1,0]]
输出:[[7,null],[13,0],[11,4],[10,2],[1,0]]
示例 2:
输入:head = [[1,1],[2,1]]
输出:[[1,1],[2,1]]
示例 3:
输入:head = [[3,null],[3,0],[3,null]]
输出:[[3,null],[3,0],[3,null]]
示例 4:
输入:head = []
输出:[]
解释:给定的链表为空(空指针),因此返回 null。
提示:
-10000 <= Node.val <= 10000
Node.random 为空(null)或指向链表中的节点。
节点数目不超过 1000 。
Answer
Idea One:
如上图所示,我们维护三个list,source list 存储head中所有块的引用;new list创建新块且与head链表块中值(val)相同;random list里是一组数字,对应head中random指向块的位置。
只要这三个list维护成功,通过循环new list,不断前向连接就能将next指针补齐;还是通过循环new list,通过利用random list储存了对应head中random指向块的位置的信息,轻松就能将new list的random指针补齐。最终返回new list中的第一个块,即为答案。时间复杂度为 O ( n 2 ) O(n^2) O(n2) 。这么大,主要是维护random list引起的。
关键是random list如何实现,也简单,且看下方代码:
"""
# Definition for a Node.
class Node:
def __init__(self, x: int, next: 'Node' = None, random: 'Node' = None):
self.val = int(x)
self.next = next
self.random = random
"""
class Solution:
def copyRandomList(self, head: 'Node') -> 'Node':
random_list = []
none_flag = -1
source_node = []
new_node = []
head1 = head
if head == None:
return None
# print (head.val,head.next,head.random)
while head != None:
source_node.append(head)
new_node.append(Node(head.val))
head = head.next
while head1 != None:
for i in range(len(source_node)):
if head1.random == None:
random_list.append(-1)
break
if source_node[i] == head1.random:
random_list.append(i)
break
head1 = head1.next
for i in range(len(random_list)):
if i <= len(random_list)-2:
new_node[i].next = new_node[i+1]
if random_list[i] == none_flag:
new_node[i].random = None
else:
new_node[i].random = new_node[random_list[i]]
return new_node[0]
Idea Two:
想法一中,时间复杂度主要是维护random list引起。对该问题,其实可通过字典快速实现。如上图,dict的key记录对应head块的引用,value则是new了一个新块且其值(val)与对应head的值相等。我们可以遍历head链表,将其块的random指针作为字典的key,这样对应字典的value不就是对应random指到的那个块吗。具体看如下代码,时间复杂度为
O
(
n
)
O(n)
O(n) 。
"""
# Definition for a Node.
class Node:
def __init__(self, x: int, next: 'Node' = None, random: 'Node' = None):
self.val = int(x)
self.next = next
self.random = random
"""
class Solution:
def copyRandomList(self, head: 'Node') -> 'Node':
if not head:
return None
copy_dict = {}
cur = head
while cur:
copy_dict[cur] = Node(cur.val)
cur = cur.next
cur = head
while cur:
if cur.next != None:
copy_dict[cur].next = copy_dict[cur.next]
else:
copy_dict[cur].next = None
if cur.random == None:
copy_dict[cur].random = None
else:
copy_dict[cur].random = copy_dict[cur.random]
cur = cur.next
return copy_dict[head]
总结
如上第二种解法也给我们启发,在查找问题中,我们可以试着使用字典,说不定就会有奇效。因为本身字典这种数据结构,在查找领域,十分快速,可以大大降低时间复杂度。