Python实现迪杰斯特拉算法过程解析
本文详细解析迪杰斯特拉(Dijkstra)算法的核心思想与Python实现过程。文章首先阐述算法基于最优子结构的贪心策略,明确其适用于无负权值的有向图。随后分步讲解算法操作流程,包括集合S与U的维护、距离更新机制及终止条件,并附带完整的Python代码示例,帮助读者深入理解单元最短路径问题的求解逻辑。
迪杰斯特拉算法思想与原理
一、 迪杰斯特拉算法思想
Dijkstra算法主要解决有向图中的单元最短路径问题,且前提条件是不存在权值为负的边。该算法在本质上类似于贪心算法,其核心依据是最短路径具有最优子结构性质。
最短路径的最优子结构性质:
若P(i,j)={Vi…Vk…Vs…Vj}表示从顶点i到j的最短路径,且k和s是该路径上的任意中间顶点,则子路径P(k,s)必然也是从k到s的最短路径。
证明:
假设P(i,j)={Vi…Vk…Vs…Vj}是从顶点i到j的最短路径,则路径长度满足P(i,j)=P(i,k)+P(k,s)+P(s,j)。如果P(k,s)不是从k到s的最短距离,则必然存在另一条更短的路径P'(k,s),使得P(i,j)=P(i,k)+P'(k,s)+P(s,j) < P(i,j),这与P(i,j)是最短路径的假设矛盾。因此,该性质得证。
基于上述性质,Dijkstra算法的具体描述如下:
Dijkstra算法描述如下:
假设存在图G=<V,E>,源顶点为V0。定义集合S={V0},distance[i]用于记录从V0到顶点i的最短距离,matrix[i][j]记录从i到j的边权值(即两点间的距离)。
1)从集合V-S中选择使dist[i]值最小的顶点i,将该顶点i加入集合S中;
2)更新与顶点i直接相邻顶点的dist值。更新公式为:dist[j]=min{dist[j], dist[i]+matrix[i][j]}。
3)重复上述过程,直到S=V,即所有顶点都包含在集合S中,算法停止。
算法具体操作步骤
二、 具体操作步骤
根据算法思想,确立的具体操作步骤如下:
(1) 初始状态下,集合S仅包含起点s;集合U包含除s外的其他所有顶点。U中各顶点的距离初始化为“起点s到该顶点的直接距离”(例如,若U中顶点v与s相邻,则距离为(s,v)的长度;若s和v不相邻,则v的距离初始化为∞)。
(2) 从集合U中选出“距离最短的顶点k”,将顶点k加入集合S中;同时,从集合U中移除顶点k。
(3) 更新集合U中各个顶点到起点s的距离。之所以需要更新U中顶点的距离,是因为上一步确定了k是求出最短路径的顶点,从而可以利用k来更新其他顶点的距离;例如,(s,v)的直接距离可能大于经过k的路径(s,k)+(k,v)的距离。
(4) 重复执行步骤(2)和(3),直到遍历完所有顶点。
Python代码实现
三、代码
def dijkstra(s, used, cost, distance, n):
distance[s] = 0
while True:
# v在这里相当于是一个哨兵,对包含起点s做统一处理!
v = -1
# 从未使用过的顶点中选择一个距离最小的顶点
for u in range(n):
if not used[u] and (v == -1 or distance[u] < distance[v]):
v = u
if v == -1:
# 说明所有顶点都维护到S中了!
break
# 将选定的顶点加入到S中, 同时进行距离更新
used[v] = True
# 更新U中各个顶点到起点s的距离。之所以更新U中顶点的距离,是由于上一步中确定了k是求出最短路径的顶点,从而可以利用k来更新其它顶点的距离;例如,(s,v)的距离可能大于(s,k)+(k,v)的距离。
for u in range(n):
distance[u] = min(distance[u], distance[v] + cost[v][u])
return distance
n, m, T = map(int, input().split())
# 标记数组:used[v]值为False说明改顶点还没有访问过,在S中,否则在U中!
used = [False for _ in range(n)]
# 距离数组:distance[i]表示从源点s到i的最短距离,distance[s]=0
distance = [float('inf') for _ in range(n)]
# cost[u][v]表示边e=(u,v)的权值,不存在时设为INF
cost = [[float('inf') for _ in range(n)] for _ in range(n)]
for _ in range(m):
e = list(map(int, input().split()))
cost[e[0] - 1][e[1] - 1] = e[2]
dis1 = dijkstra(0, used[:], cost, distance[:], n)
d1 = dis1[-1]
dis2 = dijkstra(n-1, used[:], cost, distance[:], n)
d2 = dis2[0]
print((d1+d2)*T)
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持本站。
免责声明:本文内容仅供参考,不代表本站观点。转载或引用请注明出处。









