Python实现迪杰斯特拉算法过程解析

Cora
Cora
发布时间:2026-10-14| 6人浏览过
原创

本文详细解析迪杰斯特拉(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)

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

相关标签:

免责声明:本文内容仅供参考,不代表本站观点。转载或引用请注明出处。

Python 相关产品和应用

全部下载
Selenium

Selenium

4.51.0 · Windows

用于浏览器自动化和 Web 测试的 Python 工具。

scikit-learn

scikit-learn

1.9.1 · Windows

常用 Python 机器学习算法与模型评估工具库。

Black

Black

26.10.1 · Windows

强调一致性的 Python 自动代码格式化工具。

Ruff

Ruff

0.17.1 · macOS

由 Rust 编写的高速 Python 代码检查和格式化工具。

pytest

pytest

9.1.1 · Windows

简洁而强大的 Python 自动化测试框架。

FastAPI

FastAPI

0.146.0 · Windows

高性能 Python API 框架,支持类型提示和自动接口文档。

Flask

Flask

3.1.3 · Windows

轻量灵活的 Python Web 应用框架。

Django

Django

6.1.2 · Windows

成熟的 Python Web 框架,适合构建内容站点和业务系统。

Spyder

Spyder

6.1.7 · Windows

面向科学计算与数据分析的 Python 集成开发环境。

SciPy

SciPy

1.18.1 · Windows

面向科学、工程和技术计算的 Python 算法库。