PYTHON LESSON 119
任意两点最短路:Floyd 与复杂度分析
掌握 Floyd 三重循环求任意两点最短路,学会用时间/空间复杂度权衡算法选型。
00 · 学习目标
这一课要解决什么?
学完后,你应该能够
- 三重循环:for k → for i → for j
- dist[i][k] + dist[k][j] 经过中转站 k
- INF = float("inf") 表示暂不连通
01 · 核心概念
Floyd 多源最短路与时间/空间复杂度
Floyd 多源最短路与时间/空间复杂度:Floyd 用 n×n 的矩阵存图:dist[i][j] 是 i 直达 j 的距离,自己到自己为 0,没有直达边用 float("inf") 表示无穷大。注意有向图里 dist[i][j] 与 dist[j][i] 是两回事。
GESP Python 8 级 · 多源最短路(Floyd)GESP Python 8 级 · 时间与空间复杂度分析有向无环图、拓扑排序、强连通分量和负权最短路是图论后续主题。
02 · 语法与规则
先记住这 3 条,再开始写程序
三重循环:for k → for i → for j先准确读出这条写法的结构与作用。
dist[i][k] + dist[k][j] 经过中转站 k换一组最小数据,手工推演一次结果。
INF = float("inf") 表示暂不连通再用边界值或反例确认它的适用条件。
03 · 完整实例
代码、运行结果和解释放在一起看
# 全校距离地图:Floyd 任意两点最短路
INF = float("inf")
dist = [ # dist[i][j] 表示 i 直接到 j 的距离,不通为 INF
[0, 5, INF, 10],
[INF, 0, 2, INF],
[INF, INF, 0, 3],
[INF, 1, INF, 0],
]
n = 4
for k in range(n): # 枚举“中转站”k
for i in range(n): # 枚举起点 i
for j in range(n): # 枚举终点 j
if dist[i][j] > dist[i][k] + dist[k][j]:
dist[i][j] = dist[i][k] + dist[k][j] # 经过 k 更近就更新
print("任意两点的最短距离表:")
for row in dist:
print(" ".join(str(x) for x in row))
print("三层循环,时间复杂度 O(n^3),适合点数不多的场合")任意两点的最短距离表: 0 5 7 10 inf 0 2 5 inf 4 0 3 inf 1 3 0 三层循环,时间复杂度 O(n^3),适合点数不多的场合
这是有向图:2 能到 4(距离 5),但 4 到 2 只要 1,两者不对称是正常的。
04 · 逐步理解
每一步只解决一个问题
- 01
用邻接矩阵初始化
Floyd 用 n×n 的矩阵存图:dist[i][j] 是 i 直达 j 的距离,自己到自己为 0,没有直达边用 float("inf") 表示无穷大。注意有向图里 dist[i][j] 与 dist[j][i] 是两回事。
INF = float("inf") dist = [[0, 5, INF], [INF, 0, 2], [INF, INF, 0]] - 02
三重循环:k 必须最外层
核心就三行:枚举中转站 k,再枚举起点 i 和终点 j,若 i→k→j 比直达更近就更新。k 在最外层表示“允许经过前 k 个点中转”,顺序写反答案就错——这是 Floyd 最容易踩的坑。
for k in range(n): for i in range(n): for j in range(n): - 03
松弛判断与无穷大陷阱
dist[i][j] > dist[i][k] + dist[k][j] 成立就更新。Python 的 inf 参与加法仍是 inf,比较不会出错;但在其他语言里要小心溢出。理解这一点,将来换语言也不慌。
if dist[i][j] > dist[i][k] + dist[k][j]: dist[i][j] = dist[i][k] + dist[k][j] - 04
算一笔复杂度账
时间 O(n³)、空间 O(n²)。n = 100 时 10^6 次操作,飞快;n = 1000 时 10^9 次,Python 要跑几分钟。所以 Floyd 适合 n ≤ 200 左右的稠密小图,点多了要换回 Dijkstra。
- 05
算法选型决策表
只问一个起点到所有点:Dijkstra(堆优化 O(m log n))。问任意两点之间、且点数不多:Floyd(O(n³))。要连通所有点且总代价最小:Kruskal(O(m log m))。竞赛读题第一步:先看数据范围选算法。
- 06
空间复杂度同样会挂
n×n 矩阵在 n = 10^5 时需要 10^10 个格子,内存直接爆掉。这就是为什么 Dijkstra 用邻接表(O(n+m))而不是矩阵。估算内存:每个 int 约 28 字节,列表还有额外开销,大 n 务必用表。
05 · 练习与检验
自己写出来,才算真正学会
- 1运行距离地图,核对 4×4 结果表
- 2验证 2→4 的最短路是 2→3→4 共 5
- 3给矩阵加一行一列变成 5 个点再算
- 4估算 n = 1000 时 Floyd 需要的计算量
06 · 完整知识
继续理解定义、规则和适用边界
第一次学习先完成上面的六个步骤;需要查定义、核对规则、分析误区或理解“为什么”时,再展开对应知识章。
核心数据结构树、图、DFS、BFS 与最短路从节点与边的模型出发,掌握遍历、连通性、树结构、最小生成树和最短路径的适用条件。+
正式定义
图由顶点和边构成;树是连通且无环的无向图。DFS 沿路径深入后回溯,BFS 按距离层次扩展。算法的正确选择取决于图是否有向、边权是否为负、是否稠密以及目标是遍历、连通还是最短路。
必须掌握
- 邻接矩阵占 O(V²) 空间,适合稠密图和快速查边;邻接表占 O(V+E),适合稀疏图。
- DFS 常用递归或显式栈,BFS 使用队列;一般图都必须记录 visited 防止重复和死循环。
- 无权图的 BFS 首次到达即得到最少边数距离,可用 parent 还原路径。
- 树有 V-1 条边且任意两点路径唯一;二叉树前/中/后序描述根的访问时机。
- Dijkstra 只适用于非负边权;Floyd 求所有点对最短路并允许负边,但不能有可达负环。
- Kruskal 按边权排序并用并查集避环,得到连通无向带权图的最小生成树。
常见误区
- 遍历一般图时不记录 visited
- 对负权边使用 Dijkstra
- 混淆最短路径树与最小生成树
- 递归 DFS 忽略深度限制
适用边界
- 有向无环图、拓扑排序、强连通分量和负权最短路是图论后续主题。
- 网格搜索也是图搜索:格子是顶点,可移动关系是边;不要只背二维数组模板。
Python 基础条件、循环与程序流程准确理解 if、for、while、range、break、continue 和循环嵌套,而不是背代码模板。+
正式定义
控制流决定下一条要执行的语句。分支依据布尔条件选择路径;循环在满足规则时重复执行代码块。Python 用冒号和缩进界定代码块。
必须掌握
- if/elif/else 从上到下判断,只执行第一个为真的分支;else 不写条件。
- for 依次取得可迭代对象中的元素;range(start, stop, step) 包含 start、不包含 stop,step 不能为 0。
- while 在每轮开始前检查条件;循环体必须让状态向终止条件推进。
- break 结束最内层循环,continue 跳过本轮剩余语句,循环的 else 仅在没有被 break 终止时执行。
- 嵌套循环的总执行次数通常需要把各层次数相乘;内层 break 不会结束外层循环。
- 边界测试至少覆盖空范围、单个元素、第一项命中、最后一项命中和始终不命中。
常见误区
- range 的右端点多算或少算一次
- while 忘记更新状态造成死循环
- 把两个互斥条件写成两个独立 if
- 误以为 break 会跳出所有嵌套循环
适用边界
- 流程图是算法的表示方法,不是 Python 语法。
- 递归也能表达重复,但有调用开销和递归深度限制,不能无条件代替循环。
算法方法算法、复杂度与解题验证把题意转成输入、状态、规则与输出,用正确性和复杂度共同评价解法。+
正式定义
算法是解决一类问题的有限、明确步骤。正确性说明算法对所有满足前置条件的输入都得到规定结果;时间和空间复杂度描述输入规模增长时资源使用的增长量级。
必须掌握
- 先明确输入规模 n、数据范围、目标和允许误差,再选择数据结构与算法。
- O(1)、O(log n)、O(n)、O(n log n)、O(n²)、O(2ⁿ) 表示增长量级,不是精确运行秒数。
- 顺序代码复杂度取较大项,嵌套循环常相乘,二分每步把范围缩小一半。
- 正确性可用循环不变量、数学归纳、交换论证、反证或状态定义来说明。
- 样例只验证少量输入;必须自己设计边界、极端、重复、有序/逆序和无解数据。
- 优化前先得到正确基线并测量瓶颈,不为小数据盲目增加复杂实现。
常见误区
- 只看样例通过就宣称正确
- 不看数据范围使用 O(n²)
- 二分区间开闭混用
- 把 O(n) 当成永远比 O(log n) 慢固定倍数
适用边界
- 复杂度隐藏常数与硬件差异,但仍是比较规模增长的核心工具。
- 考场策略、课程完成度和算法能力是不同证据,任何单项都不能保证考级通过。
工程能力异常、文件、测试与调试读懂报错、缩小问题、设计测试,并安全地打开、读取和关闭文本文件。+
正式定义
异常是在运行期间表示错误或特殊情况的对象。调试是用可复现输入和证据定位实际行为与预期行为差异的过程;文件对象连接程序与持久化字节数据。
必须掌握
- 先读 traceback 最后一行的异常类型与消息,再从最靠近自己代码的栈帧向上追踪。
- try 只包可能失败的最小代码;except 捕获具体异常;else 处理成功路径;finally 做必需清理。
- raise 主动报告不满足的前置条件;assert 用于开发期内部假设,不用于校验不可信用户输入。
- with open(...) as file 会在退出代码块时可靠关闭文件。文本模式必须明确编码,本站统一推荐 encoding='utf-8'。
- 测试至少包含正常值、边界值、空数据、极端值和反例;每个测试只应有明确目的。
- 定位错误时一次只改一个假设,保留能稳定复现问题的最小输入。
常见误区
- 使用 except: 吞掉所有错误
- 只测题目样例就认为程序正确
- 文本文件不写 encoding
- 修复报错表象却不验证根因
适用边界
- 在线判题的学生代码由独立 Worker 执行;文件系统、网络和资源权限必须受平台限制。
- 二进制文件、JSON/CSV 和数据库各有专门格式与错误处理方式,不能按普通文本随意拆分。
完成检查