找到你联系人中的芒果商——广度优先搜索算法(BFS)实例Demo

参考:《算法图解》(人民邮电出版社)

六度关系理论与寻找芒果商

每个人都有自己的关系网,假如下面是你微信好友的关系网,其中you节点代表你自己:

you直接相连的联系人是直接好友,比如图中的Tony,你可以通过一步就联系到Tony。而对于直接好友的好友,比如图中的Alice,Alice虽然是Tony的好友,但不是你的好友,你无法直接联系到Alice,只能通过直接好友Tony作为桥梁才能联系到Alice。

我们把图中的像Tony、Tom、John这样的直接好友称为一度关系,也就是下图中绿色圆圈中的好友。而下图中蓝色圆圈和黄色圆圈中的好友则分别是二度关系三度关系

有一种理论讲到,你和世界上任何人之间最多只隔了六度关系,换句话说,你最多通过六个人就可以联系到这世界上的任何人。

暂且不去争论这个理论的对错,我们下面来考虑一个实际问题:假如你想找到你的关系网中和你关系最近的芒果商有谁?该怎么找呢?(这里假定名字以大写或小写的字母M开头的人就是芒果商,虽然比较搞笑,但为了简单起见先这么定)一种可行的思路就是采用广度优先搜索的策略。

既然是想找到关系最近的芒果商,那肯定优先找一度关系中的联系人比较好,一度关系联系人找了个遍之后,如果没有发现任何人是芒果商,那接着去找二度联系人、三度联系人、...、N度联系人。

在算法实现的过程中,查找过程中最好用队列去存储待找的联系人,一开始把这个队列初始化为开始节点,每找一个联系人之后,就把这个联系人的所有直接联系人添加到队列的尾部,供后面查找。此外,为了避免重复查找,还需要创建一个已经处理过的联系人列表,每处理完一个联系人后都把他都加入到这个列表中。

下面是详细的算法实现,几乎每一行都写了详细的注释,还是比较好理解的:)

算法实现:找到芒果商

# coding:utf-8
# 广度优先搜索寻找图中关系最近的芒果商
# BFS特点:地毯式稳步推进,不留死角。
def BFS(graph,start):
    '''
    广度优先搜索查找在图graph中从start节点到goal节点之间的最短路径列表和所有的可达路径
    
    :param graph: 图的散列表结构对象
    :param start: 开始节点
    :return: 离start节点最近的芒果商,如果有多个芒果商都最近,则返回找到的第一个;如果没有找到芒果商,则返回None
    '''
    # 定义一个节点队列,用于存储待查找的节点,初始化为[start],在查找的过程中会不断更新该队列
    search_queue = [start]
    # 已经处理过的节点的列表,防止重复查找
    searched = []
    # 只要search_queue队列不为空,就查找下去
    while search_queue:
        # 从search_queue队列出队一个节点
        node = search_queue.pop(0)
        # 如果node是芒果商,则返回
        if is_mangoman(node):
            return node
        
        # 遍历node节点的所有邻居节点,把每个邻居节点分别添加到search_queue队列中去
        for neighbor_node in graph.get(node,[]):
            # 如果neighbor_node不在searched列表中,则将其入队
            if neighbor_node not in searched:
                search_queue.append(neighbor_node)
    return None

def is_mangoman(name):
    '''
    根据一个人名判断一个人是否是芒果商
    
    :param name: 人名
    :return: True或False
    '''
    if not name:
        return False
    # 名字以'm'或'M'开头就认为是芒果商(虽然这确实有点搞笑)
    if name.startswith('m') or name.startswith('M'):
        return True
    else:
        return False
    
def main():
    # 构建图,有一个前提假设:图中的姓名没有重复的(保证唯一性)
    graph = {
        'you':['Tony','John','Tom'],
        'Tony':['Alice','Andy'],
        'John':['Nancy','Ben'],
        'Tom':['Andy','Mike'],
        'Andy':['Mike','Mercy'],
        'Nancy':['Ben'],
        'Ben':['Tom']
    }
    start = 'you'
    mangoman = BFS(graph,start)
    print 'the closest mangoman to {0} is: {1}'.format(start,mangoman)

if __name__ == '__main__':
    main()
    '''
    输出:
    the closest mangoman to you is: Mike
    '''

算法实现进阶:找到能够联系到芒果商的最短路径

上面的算法只是找到了和你关系最近的芒果商,但是并没有得出通过什么路径可以联系到这个联系人,下面优化算法,找到这样的路径:

# coding:utf-8
# 广度优先搜索求图中的最短路径:寻找芒果商
# BFS特点:地毯式稳步推进,不留死角。
def BFS(graph,start,goal):
    '''
    广度优先搜索查找在图graph中从start节点到goal节点之间的最短路径列表和所有的可达路径
    
    :param graph: 图的散列表结构对象
    :param start: 开始节点
    :param goal: 目标节点
    :return: 从start节点到goal节点之间的最短路径列表和所有的可达路径列表
    '''
    # 定义一个路径队列,用于存储可能的路径,初始化为[[start]],在查找的过程中会不断更新该队列
    paths_queue = [[start]]
    # 所有从start到goal可达的路径的列表
    all_paths = []
    # 只要paths_queue队列不为空,就查找下去
    while paths_queue:
        # 从paths_queue队列出队一个路径
        path = paths_queue.pop(0)
        # 获取该路径中的最后一个节点
        last_node = path[-1]
        # 如果last_node等于goal,说明当前路径path就是一条从start到goal可达的路径
        if last_node == goal:
            # 如果当前路径不在all_paths中,则把当前路径path添加到all_paths列表中
            if path not in all_paths:
                all_paths.append(path)
        
        # 遍历last_node节点的所有邻居节点,把每个邻居节点分别添加到path后面形成新的路径,添加到路径队列中去
        for neighbor_node in graph.get(last_node,[]):
            # 创建一个新的路径对象,防止内存覆盖
            new_path = list(path)
            # 添加邻居节点到new_path中形成一条新的路径
            new_path.append(neighbor_node)
            # 如果new_path不在队列中,则将其入队
            if new_path not in paths_queue:
                paths_queue.append(new_path)
    # 从all_paths中过滤出最短路径列表
    shortest_paths_len = len(min(all_paths,key = lambda path:len(path)))
    shortest_paths = [path for path in all_paths if len(path) == shortest_paths_len]
    return shortest_paths,all_paths
    
def output(paths):
    '''
    更好地输出路径
    
    :param paths: 路径列表,每条路径都是一个节点列表
    :return: 一个路径字符串
    '''
    for path in paths:
        print '->'.join(path)

def main():
    # 构建图,有一个前提假设:图中的姓名没有重复的(保证唯一性)
    graph = {
        'you':['Tony','John','Tom'],
        'Tony':['Alice','Andy'],
        'John':['Nancy','Ben'],
        'Tom':['Andy','Mike'],
        'Andy':['Mike','Mercy'],
        'Nancy':['Ben'],
        'Ben':['Tom']
    }
    start,goal = 'you','Mike'
    shortest_paths,all_paths = BFS(graph,start,goal)
    print '>>> shortest_paths from {0} to {1}:'.format(start,goal)
    output(shortest_paths)
    
    print '\n'
    print '>>> all_paths from {0} to {1}:'.format(start,goal)
    output(all_paths)

if __name__ == '__main__':
    main()
    '''
    输出:
    >>> shortest_paths from you to Mike:
    you->Tom->Mike

    >>> all_paths from you to Mike:
    you->Tom->Mike
    you->Tony->Andy->Mike
    you->Tom->Andy->Mike
    you->John->Ben->Tom->Mike
    you->John->Nancy->Ben->Tom->Mike
    you->John->Ben->Tom->Andy->Mike
    you->John->Nancy->Ben->Tom->Andy->Mike
    '''

BFS算法时间复杂度

O(V+E)
其中V为图中的节点数,E为图中的边数。

BFS算法特点一句话总结

地毯式稳步推进,不留死角。

总结

本文中通过BFS算法找到了你的关系网中离你最近的芒果商,一开始只是找到了离你最近的芒果商,而并不知道路径该怎么走;后来进一步又优化了算法,找到了能够联系到芒果商的最短路径。


代码已经更新至:我的GitHub

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,444评论 6 496
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,421评论 3 389
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 160,036评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,363评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,460评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,502评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,511评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,280评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,736评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,014评论 2 328
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,190评论 1 342
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,848评论 5 338
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,531评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,159评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,411评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,067评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,078评论 2 352