例题建议 45 分钟学习等级 8/8

PYTHON LESSON 124

例题精练:基站选址的最坏情况

拆解“最小化最大距离”题型:先 Floyd 求全源最短路,再逐点取最大值、全局取最小。

00 · 学习目标

这一课要解决什么?

先想一想基站建在地图正中间就一定最好吗?“最坏情况最好”到底怎么算?
完成任务通信基站选址官
学习顺序读题 → 列出已知与目标 → 选择方法 → 编码 → 验证

学完后,你应该能够

  • Floyd 三重循环 for k → for i → for j
  • worst[v] = max(dist[v]) 每个点的最坏情况
  • 严格小于才更新答案,编号小的自动优先

01 · 核心概念

Floyd 全源最短路 + 最小化最大值(minimax)

Floyd 全源最短路 + 最小化最大值(minimax):“最坏情况最好”要读两遍:对每个候选村庄,它的代价是“到最远村庄的最短距离”(最坏情况);然后在所有村庄里挑这个代价最小的(最好)。把两层意思拆开,代码结构就出来了:外层挑村,内层算最远。

考级对应GESP Python 8 级 · 多源最短路(Floyd)GESP Python 8 级 · 时间与空间复杂度分析
学习边界

有向无环图、拓扑排序、强连通分量和负权最短路是图论后续主题。

02 · 语法与规则

先记住这 3 条,再开始写程序

01Floyd 三重循环 for k → for i → for j

先准确读出这条写法的结构与作用。

02worst[v] = max(dist[v]) 每个点的最坏情况

换一组最小数据,手工推演一次结果。

03严格小于才更新答案,编号小的自动优先

再用边界值或反例确认它的适用条件。

03 · 完整实例

代码、运行结果和解释放在一起看

g8-base-station-cracked.pyPYTHON 3.12
# 例题精练:基站选址(题库 py-base-station 同款思路)
# 选一个村庄建基站,使它到其他村庄最短距离的最大值尽量小
import sys

def main():
    data = sys.stdin.buffer.read().split()
    pos = 0
    n = int(data[pos]); m = int(data[pos + 1]); pos += 2
    INF = float("inf")
    dist = [[INF] * (n + 1) for _ in range(n + 1)]
    for i in range(1, n + 1):
        dist[i][i] = 0
    for _ in range(m):
        a = int(data[pos]); b = int(data[pos + 1]); w = int(data[pos + 2]); pos += 3
        if w < dist[a][b]:              # 可能有重边,只留最短的
            dist[a][b] = w
            dist[b][a] = w
    # Floyd:三重循环求任意两点最短路
    for k in range(1, n + 1):
        dk = dist[k]
        for i in range(1, n + 1):
            di = dist[i]
            if di[k] == INF:
                continue                # i 到不了 k,这轮跳过
            via = di[k]
            for j in range(1, n + 1):
                nd = via + dk[j]
                if nd < di[j]:
                    di[j] = nd
    # 每个村庄的“最坏情况”:到其他村庄的最大距离
    best_v, best_worst = -1, INF
    for v in range(1, n + 1):
        cur = max(dist[v][1:])
        if cur < best_worst:            # 严格小于,编号小的自动优先
            best_worst = cur
            best_v = v
    print(best_v)
    print(best_worst)

main()
运行结果OUTPUT
2
5

假设输入为:第一行 4 5(4 村 5 路),随后 5 行依次是 1 2 2 / 1 3 3 / 2 3 4 / 2 4 5 / 3 4 6。输出两行:基站建在 2 号村,最坏距离 5。

在新标签运行和修改这个实例已装入本课代码 · 可自定义输入 · 可提交判题

04 · 逐步理解

每一步只解决一个问题

  1. 01

    第一步:读题,读懂 minimax

    “最坏情况最好”要读两遍:对每个候选村庄,它的代价是“到最远村庄的最短距离”(最坏情况);然后在所有村庄里挑这个代价最小的(最好)。把两层意思拆开,代码结构就出来了:外层挑村,内层算最远。

    PYTHON
    # 外层:枚举建基站的村 v
    # 内层:max(dist[v]) 是 v 的最坏情况
  2. 02

    第二步:看数据范围选算法

    n ≤ 150,Floyd 的 O(n³) 约 3.4×10^6 次操作,Python 也能秒过;而且这题要的是“任意两点”的距离表,Floyd 一次算全最划算。若 n 到 10^4 就得改成跑 n 次堆优化 Dijkstra——先估算再动手的老规矩不变。

    PYTHON
    # n ≤ 150 → O(n^3) ≈ 3.4×10^6,Floyd 可行
  3. 03

    第三步:建图的两个细节

    邻接矩阵初始化:对角线 0、其余 INF。输入可能有重边(两个村之间给了两条路),只保留最短的那条——不重边判断就会用更长的边算距离,答案直接偏大,这是竞赛输入常埋的坑。

    PYTHON
    if w < dist[a][b]:
        dist[a][b] = w
        dist[b][a] = w
  4. 04

    第四步:Floyd 主体与一个小优化

    三重循环 k→i→j 照模板写,k 必须最外层。顺手加一个优化:di[k] == INF 时整轮跳过——i 根本到不了中转站 k,内层循环纯属浪费。稀疏图上这行 continue 能省下可观的时间。

    PYTHON
    if di[k] == INF:
        continue
  5. 05

    第五步:扫描答案,编号小的优先

    逐行取 max 得到每个村庄的最坏情况,再取其中的最小值。比较用严格小于 cur < best_worst:打平时不更新,先扫到的编号小者自动留下——正好满足题目“并列取编号最小”的要求,不用额外写 tie-break。

    PYTHON
    if cur < best_worst:
        best_worst = cur
        best_v = v
  6. 06

    变体与复盘

    把“最远村庄距离最小”改成“到所有村庄总距离最小”,就得到另一个经典考点(图的重心):只需把 max 换成 sum。先全源最短路、再按行统计,是所有选址类题目的通用骨架。

    PYTHON
    cur = sum(dist[v][1:])  # 变体:总距离最小

05 · 练习与检验

自己写出来,才算真正学会

本课实作步骤
  1. 1运行选址官,核对建在 2 号村、最坏距离 5
  2. 2手算 1 号村的最坏距离为什么是 7
  3. 3把 3-4 的路改成 1,看选址是否改变
  4. 4解释这题为什么选 Floyd 而不是 4 次 Dijkstra
打开本课编程实验室编辑、运行、测试、判题都在一个页面完成

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) 慢固定倍数

适用边界

  • 复杂度隐藏常数与硬件差异,但仍是比较规模增长的核心工具。
  • 考场策略、课程完成度和算法能力是不同证据,任何单项都不能保证考级通过。
打开本章完整示例与独立阅读页 →
Python 基础运算符、表达式与优先级完整区分算术、比较、逻辑、成员、身份和位运算,并用优先级表消除歧义。

正式定义

表达式求值得到一个值。运算符规定如何组合操作数;当一个表达式含多个运算符时,优先级和结合方向决定求值顺序,括号可以明确改变顺序。

必须掌握

  • / 总是得到浮点结果;// 是向负无穷方向取整的整除;% 与 // 满足 a == (a // b) * b + a % b。
  • 比较可以链式书写,如 0 <= x < 10;and/or 会短路并返回最后求值的操作数,不一定返回 bool。
  • == 比较值是否相等,is 比较是否为同一个对象;判断 None 应写 is None。
  • in/not in 做成员测试;对 dict 测试的是键。
  • 位运算作用于整数的二进制位;负整数按无限长二进制补码语义理解。
  • 复杂表达式即使能靠优先级正确运行,也应使用括号表达意图。

常见误区

  • 把 // 当成简单截断
  • 用 is 比较数字或字符串的值
  • 忘记 and 的优先级高于 or
  • 连续位移、比较和逻辑运算却不加括号

适用边界

  • 浮点数比较受二进制表示误差影响,需要按问题选择容差。
  • 运算符可由自定义类重载,因此相同符号对不同类型可能有不同语义。
打开本章完整示例与独立阅读页 →

完成检查

确认自己会解释、会编写、会验证