算法快学笔记(十二):图的广度优先搜索(BFS-Breadth First Search)

1. 介绍

广度优先搜索(BFS)是一个经典的图算法,该算法能够找出两样东西之间的最短距离!使用广度优先搜索可以:

  • 编写国际跳棋AI,计算最少走多少步就可获胜;
  • 编写拼写检查器,计算最少编辑多少个地方就可将错拼的单词改成正确的单词,如将READED改为READER需要编辑一个地方;
  • 根据你的人际关系网络找到关系最近的医生。

要说明的是,广度优先搜索是一种用于图的查找算法,可帮助
回答两类问题。

  • 第一类问题:从节点A出发,有前往节点B的路径吗?
  • 第二类问题:从节点A出发,前往节点B的哪条路径最短?

2. 原理说明

假设你经营着一个芒果农场,需要寻找芒果销售商,以便将芒果卖给他。在Facebook,你与芒果销售商有联系吗?为此,你可在朋友中查找,这种查找很简单。首先,创建一个朋友名单,假设你没有朋友是芒果销售商,那么你就必须在朋友的朋友中查找,检查名单中的每个人时,你也都将其朋友加入名单

算法快学笔记(十二):图的广度优先搜索(BFS-Breadth First Search)_第1张图片

这样一来,你不仅在朋友中查找,还在朋友的朋友中查找。别忘了,你的目标是在你的人际关系网中找到一位芒果销售商。因此,如果Alice不是芒果销售商,就将其朋友也加入到名单中。这意味着你将在她的朋友、朋友的朋友等中查找。使用这种算法将搜遍你的整个人际关系网,直到找到芒果销售商。这就是广度优先搜索算法。

对于BFS要解决的第一个问题很容易处理,只要遍历朋友列表,就能得知朋友圈中到底有没有芒果销售商,下面来看看如何找到哪个销售商和你的关系是最好的。

2.1 查找最亲密的销售商

为了便于度量关系的亲密度,我们假设朋友是一度关系,朋友的朋友是二度关系…,一度关系胜过二度关系,二度关系胜过三度关系,以此类推.

为了确保找到的销售商关系是最亲密的,你应先在一度关系中搜索,确定其中没有芒果销售商后,才在二度关系中搜索。

BFS算法的执行流程也就是**搜索范围从起点开始逐渐向外延伸,即先检查一度关系,再检查二度关系。**

在查找列表的过程中,有点需要注意,因为1度关系比二度关系先添加,二度关系比三度关系先添加,因此只有按添加顺序查找,才能找到最短路径。此处可以使用队列实现。

3. 算法实现

# -*- coding:utf-8 -*-
# @Author:sunaihua 
from collections import deque

graph = {}
graph["you"] = ["alice", "bob", "claire"]
graph["bob"] = ["anuj", "peggy"]
graph["alice"] = ["peggy"]
graph["claire"] = ["thom", "jonny"]
graph["anuj"] = []
graph["peggy"] = []
graph["thom"] = []
graph["jonny"] = ["bob"]

# 节点对象,用来标识各个节点之间的父子关系
# name:字符串类型,代表节点名称
# parent:Node类型
class Node:
    def __init__(self, name, parent):
        self.name = name
        self.parent = parent

    def __str__(self):
        return self.name


def is_seller(friend):
    return friend.name[-1] == 'm'

# 通过bfs找到最近的节点
def bfs_search():
    search_queue = deque()
    # 初始化任务队列
    search_queue += [Node(name=friend, parent=Node(name='you', parent=None)) for friend in graph['you']]
    searchrd = []
    while search_queue:
        # 每次都弹出队列最左面的元素,且是关系最亲密的。
        friend = search_queue.popleft()
        # 防止重复搜索
        if friend not in searchrd:
            if is_seller(friend):
                print "has seller,%s is seller" % (friend)
                return friend

            else:
                # 将当前friend的所有friend添加到队列的最右面
                search_queue += [Node(name=f, parent=friend) for f in graph[friend.name]]
    return None

# 根据超找到的对象,打印路径
def print_path(node):
    node_name_stack = []
    node_name_stack.append(node.name)
    parent_node = node.parent
    # 生成节点名称列表
    while parent_node:
        node_name_stack.append(parent_node.name)
        parent_node = parent_node.parent
    # 反转列表
    node_name_stack.reverse()
    # 打印路径
    print "friends path:","--->".join(node_name_stack)


if __name__ == '__main__':
    target = bfs_search()
    print_path(target)


4. 运行时间

如果你在你的整个人际关系网中搜索芒果销售商,就意味着你将沿每条边前行(记住,边是
从一个人到另一个人的箭头或连接),因此运行时间至少为O(边数)。
你还使用了一个队列,其中包含要检查的每个人。将一个人添加到队列需要的时间是固定的,即为O(1),因此对每个人都这样做需要的总时间为O(人数)。所以,广度优先搜索的运行时间为
O(人数 + 边数),这通常写作O(V + E),其中V为顶点(vertice)数,E为边数

5. 总结

  • 广度优先搜索主要用来解决最短路径问题
  • FS算法的执行流程是搜索范围从起点开始逐渐向外延伸,即先检查一度关系,再检查二度关系…
  • 时间复杂度为O(V + E)

你可能感兴趣的:(算法快学,Algorithm,趣学算法与数据结构)