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

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

作者: m2fox | 来源:发表于2018-05-15 13:31 被阅读0次

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

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

    每个人都有自己的关系网,假如下面是你微信好友的关系网,其中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

    相关文章

      网友评论

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

          本文链接:https://www.haomeiwen.com/subject/frisdftx.html