题目

复杂链表的复制

复杂链表可以看成是每个节点为顶点，每个节点初度最大为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)