python Dijkstra算法实现最短路径问题的方法


Posted in Python onSeptember 19, 2019

本文借鉴于张广河教授主编的《数据结构》,对其中的代码进行了完善。

从某源点到其余各顶点的最短路径

Dijkstra算法可用于求解图中某源点到其余各顶点的最短路径。假设G={V,{E}}是含有n个顶点的有向图,以该图中顶点v为源点,使用Dijkstra算法求顶点v到图中其余各顶点的最短路径的基本思想如下:

  • 使用集合S记录已求得最短路径的终点,初始时S={v}。
  • 选择一条长度最小的最短路径,该路径的终点w属于V-S,将w并入S,并将该最短路径的长度记为Dw。
  • 对于V-S中任一顶点是s,将源点到顶点s的最短路径长度记为Ds,并将顶点w到顶点s的弧的权值记为Dws,若Dw+Dws<Ds,
  • 则将源点到顶点s的最短路径长度修改为Dw+Ds=ws。
  • 重复执行2和3,知道S=V。
  • 为了实现算法,
  • 使用邻接矩阵Arcs存储有向网,当i=j时,Arcs[i][j]=0;当i!=j时,若下标为i的顶点到下标为j的顶点有弧且弧的权值为w,则Arcs[i][j]=w,否则Arcs[i][j]=float(‘inf')即无穷大。
  • 使用Dist存储源点到每一个终点的最短路径长度。
  • 使用列表Path存储每一条最短路径中倒数第二个顶点的下标。
  • 使用flag记录每一个顶点是否已经求得最短路径,在思想中即是判断顶点是属于V集合,还是属于V-S集合。

代码实现

#构造有向图Graph
class Graph:
  def __init__(self,graph,labels): #labels为标点名称
    self.Arcs=graph
    self.VertexNum=graph.shape[0]
    self.labels=labels
def Dijkstra(self,Vertex,EndNode): #Vertex为源点,EndNode为终点
  Dist=[[] for i in range(self.VertexNum)] #存储源点到每一个终点的最短路径的长度
  Path=[[] for i in range(self.VertexNum)] #存储每一条最短路径中倒数第二个顶点的下标
  flag=[[] for i in range(self.VertexNum)] #记录每一个顶点是否求得最短路径
  index=0
  #初始化
  while index<self.VertexNum:
    Dist[index]=self.Arcs[Vertex][index]
    flag[index]=0
    if self.Arcs[Vertex][index]<float('inf'): #正无穷
      Path[index]=Vertex
    else:
      Path[index]=-1 #表示从顶点Vertex到index无路径
    index+=1
  flag[Vertex]=1
  Path[Vertex]=0
  Dist[Vertex]=0
  index=1
  while index<self.VertexNum:
    MinDist=float('inf')
    j=0
    while j<self.VertexNum:
      if flag[j]==0 and Dist[j]<MinDist:
        tVertex=j #tVertex为目前从V-S集合中找出的距离源点Vertex最断路径的顶点
        MinDist=Dist[j]
      j+=1
    flag[tVertex]=1
    EndVertex=0
    MinDist=float('inf') #表示无穷大,若两点间的距离小于MinDist说明两点间有路径
    #更新Dist列表,符合思想中第三条
    while EndVertex<self.VertexNum:
      if flag[EndVertex]==0:
        if self.Arcs[tVertex][EndVertex]<MinDist and Dist[
          tVertex]+self.Arcs[tVertex][EndVertex]<Dist[EndVertex]:
          Dist[EndVertex]=Dist[tVertex]+self.Arcs[tVertex][EndVertex]
          Path[EndVertex]=tVertex
      EndVertex+=1
    index+=1
  vertex_endnode_path=[] #存储从源点到终点的最短路径
  return Dist[EndNode],start_end_Path(Path,Vertex,EndNode,vertex_endnode_path)
#根据本文上述定义的Path递归求路径
def start_end_Path(Path,start,endnode,path):
  if start==endnode:
    path.append(start)
  else:
    path.append(endnode)
    start_end_Path(Path,start,Path[endnode],path)
  return path

if __name__=='__main__':
  #float('inf')表示无穷
  graph=np.array([[0,6,5,float('inf'),float('inf'),float('inf')],
          [float('inf'),0,2,8,float('inf'),float('inf')],
          [float('inf'),float('inf'),0,float('inf'),3,float('inf')],
          [float('inf'),float('inf'),7,0,float('inf'),9],
          [float('inf'),float('inf'),float('inf'),float('inf'),0,9],
          [float('inf'),float('inf'),float('inf'),float('inf'),0]])
  G=Graph(graph,labels=['a','b','c','d','e','f'])
  start=input('请输入源点')
  endnode=input('请输入终点')
  dist,path=Dijkstra(G,G.labels.index(start),G.labels.index(endnode))
  Path=[]
  for i in range(len(path)):
    Path.append(G.labels[path[len(path)-1-i]])
  print('从顶点{}到顶点{}的最短路径为:\n{}\n最短路径长度为:{}'.format(start,endnode,Path,dist))

输出结果如下:

请输入源点
a
请输入终点
f
从顶点a到顶点f的最短路径为:
['a', 'c', 'e', 'f']
最短路径长度为:17

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python文件和目录操作方法大全(含实例)
Mar 12 Python
利用Python演示数型数据结构的教程
Apr 03 Python
在Python上基于Markov链生成伪随机文本的教程
Apr 17 Python
基于Python实现通过微信搜索功能查看谁把你删除了
Jan 27 Python
python生成式的send()方法(详解)
May 08 Python
python PyTorch预训练示例
Feb 11 Python
caffe binaryproto 与 npy相互转换的实例讲解
Jul 09 Python
解决Python对齐文本字符串问题
Aug 28 Python
python matplotlib:plt.scatter() 大小和颜色参数详解
Apr 14 Python
Python虚拟环境venv用法详解
May 25 Python
keras的siamese(孪生网络)实现案例
Jun 12 Python
一篇文章弄懂Python中的内建函数
Aug 07 Python
PHP统计代码行数的小代码
Sep 19 #Python
Python实现代码统计工具
Sep 19 #Python
python实现统计代码行数的小工具
Sep 19 #Python
python日志模块logbook使用方法
Sep 19 #Python
python统计指定目录内文件的代码行数
Sep 19 #Python
python如何从文件读取数据及解析
Sep 19 #Python
python实现代码统计器
Sep 19 #Python
You might like
PHP基础教程(php入门基础教程)一些code代码
2013/01/06 PHP
PHP使用fopen与file_get_contents读取文件实例分享
2016/03/04 PHP
基于JQuery的cookie插件
2010/04/07 Javascript
javascript下string.format函数补充
2010/08/24 Javascript
js日历功能对象
2012/01/12 Javascript
Js制作简单弹出层DIV在页面居中 中间显示遮罩的具体方法
2013/08/08 Javascript
node.js中的http.response.getHeader方法使用说明
2014/12/14 Javascript
jQuery基于图层模仿五星星评价功能的方法
2015/05/07 Javascript
javascript比较两个日期相差天数的方法
2015/07/24 Javascript
微信小程序 action-sheet底部菜单详解
2016/10/27 Javascript
JS实现的适合做faq或menu滑动效果示例
2016/11/17 Javascript
jQuery弹出窗口打开链接的实现代码
2016/12/24 Javascript
Bootstrap导航条学习使用(一)
2017/02/08 Javascript
vue实现文章内容过长点击阅读全文功能的实例
2017/12/28 Javascript
在vue中通过axios异步使用echarts的方法
2018/01/13 Javascript
nodejs图片处理工具gm用法小结
2018/12/12 NodeJs
微信小程序顶部导航栏滑动tab效果
2019/01/28 Javascript
JavaScript数据结构与算法之二叉树添加/删除节点操作示例
2019/03/01 Javascript
微信小程序 腾讯地图显示偏差问题解决
2019/07/27 Javascript
VUE兄弟组件传值操作实例分析
2019/10/26 Javascript
vue移动端模态框(可传参)的实现
2019/11/20 Javascript
Nuxt页面级缓存的实现
2020/03/09 Javascript
如何HttpServletRequest文件对象并储存
2020/08/14 Javascript
[05:46]DOTA2英雄梦之声_第18期_陈
2014/06/20 DOTA
在Python中用has_key()方法查找键是否存在的教程
2015/05/21 Python
在Python中处理日期和时间的基本知识点整理汇总
2015/05/22 Python
django使用django-apscheduler 实现定时任务的例子
2019/07/20 Python
Python3实现将一维数组按标准长度分隔为二维数组
2019/11/29 Python
你需要学会的8个Python列表技巧
2020/06/24 Python
经理秘书找工作求职信
2013/12/19 职场文书
装修五一活动策划案
2014/01/23 职场文书
大学生毕业自我鉴定范文
2014/02/03 职场文书
党建工作汇报材料
2014/12/24 职场文书
工作失误检讨书范文
2015/01/26 职场文书
敬老院义诊活动总结
2015/05/07 职场文书
护理工作心得体会
2016/01/22 职场文书