1. 程式人生 > 程式設計 >Python實現迪傑斯特拉演算法並生成最短路徑的示例程式碼

Python實現迪傑斯特拉演算法並生成最短路徑的示例程式碼

def Dijkstra(network,s,d):#迪傑斯特拉演算法算s-d的最短路徑,並返回該路徑和代價
  print("Start Dijstra Path……")
  path=[]#s-d的最短路徑
  n=len(network)#鄰接矩陣維度,即節點個數
  fmax=999
  w=[[0 for i in range(n)]for j in range(n)]#鄰接矩陣轉化成維度矩陣,即0→max
  book=[0 for i in range(n)]#是否已經是最小的標記列表
  dis=[fmax for i in range(n)]#s到其他節點的最小距離
  book[s-1]=1#節點編號從1開始,列表序號從0開始
  midpath=[-1 for i in range(n)]#上一跳列表
  for i in range(n):
    for j in range(n):
      if network[i][j]!=0:
        w[i][j]=network[i][j]#0→max
      else:
        w[i][j]=fmax
      if i==s-1 and network[i][j]!=0:#直連的節點最小距離就是network[i][j]
        dis[j]=network[i][j]
  for i in range(n-1):#n-1次遍歷,除了s節點
    min=fmax
    for j in range(n):
      if book[j]==0 and dis[j]<min:#如果未遍歷且距離最小
        min=dis[j]
        u=j
    book[u]=1
    for v in range(n):#u直連的節點遍歷一遍
      if dis[v]>dis[u]+w[u][v]:
        dis[v]=dis[u]+w[u][v]
        midpath[v]=u+1#上一跳更新
  j=d-1#j是序號
  path.append(d)#因為儲存的是上一跳,所以先加入目的節點d,最後倒置
  while(midpath[j]!=-1):
    path.append(midpath[j])
    j=midpath[j]-1
  path.append(s)
  path.reverse()#倒置列表
  print(path)
  #print(midpath)
  print(dis)
  #return path

network=[[0,1,2,0],[1,4,3,[0,4],[2,6,2],0]]
Dijkstra(network,6)

以上就是Python實現迪傑斯特拉演算法並生成最短路徑的示例程式碼的詳細內容,更多關於Python實現迪傑斯特拉演算法的資料請關注我們其它相關文章!