python 广度优先搜索模板 转

# %%
# 模板1,不带hash表，每个节点可以访问多次，遍历树没有问题，遍历图就会陷入循环
def BFS1(root, target):
    # 记录遍历的层次
    step = 0
    # 这一层的节点列表，注意第一层要将根节点放进去
    this_layer = [root]
    # 当这一层的节点列表不为空，就要一直遍历
    while this_layer != []:
        # 下一层的节点列表
        next_layer = []
        # 遍历这一层的节点
        for node in this_layer:
            if node.value == target:
                return step  # 找到了就返回
            # 同时要把每个节点的下一个节点加入到下一层节点列表
            next_layer.append(node.next)
        # 遍历完这一层的节点，step就加一
        step += 1
        this_layer = next_layer


# %%
# 模板2，带hash表，保证每个节点只访问一次，遍历图没有问题
def BFS2(root, target):
    # 只需要在BFS1的基础上将遍历过的节点加入hash表就好了
    hash_table = set()
    # 记录遍历的层次
    step = 0
    # 这一层的节点列表，注意第一层要将根节点放进去
    this_layer = [root]
    # 当这一层的节点列表不为空，就要一直遍历
    while this_layer != []:
        # 下一层的节点列表
        next_layer = []
        # 遍历这一层的节点
        for node in this_layer:
            # 没遍历过才需要遍历
            if node not in hash_table:
                if node.value == target:
                    return step  # 找到了就返回
                # 被遍历了就加入hash_table
                hash_table.add(node)
            # 同时要把每个节点的下一个节点加入到下一层节点列表
            next_layer.append(node.next)
        # 遍历完这一层的节点，step就加一
        step += 1
        this_layer = next_layer

作者：intA
链接：https://leetcode-cn.com/leetbook/read/queue-stack/kc5ge/?discussion=xevkAA
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。