题目
复杂链表的复制
复杂链表可以看成是每个节点为顶点,每个节点初度最大为2的图。例如下列这个复杂链表:
对于图的遍历,有广度搜索和深度搜索两种遍历方式。
复杂链表的定义:
class ComplexListNode():
def __init__(self, val=None, next=None, random=None):
self.val = val
self.next = next
self.random = random
一、思路一 深度搜索
深度搜索。
def copyRandomList(head):
def dfs(head):
if not head:
return None
if head in visited:
return visited[head]
copy = ComplexListNode(head.val, None, None)
visited[head] = copy
copy.next = dfs(head.next)
copy.random = dfs(head.random)
return copy
visited = {}
return dfs(head)
二、思路二 广度搜索
def copy_random_list_2(head):
'''广度搜索
'''
visited = {}
def bfs(head):
if not head:
return None
queue = []
queue.append(head)
copy = ComplexListNode(head.val, None, None)
visited[head] = copy
while queue:
node = queue.pop(0)
if node.next and node.next not in visited:
visited[node.next] = ComplexListNode(node.next.val, [], [])
queue.append(node.next)
if node.random and node.random not in visited:
visited[node.random] = ComplexListNode(node.random.val, [], [])
queue.append(node.random)
visited[node].next = visited.get(node.next)
visited[node].random = visited.get(node.random)
return bfs(head)