小红需要为模型训练选择流水线并行度 p 。合法的 p 必须满足: p 不超过硬件卡数;模型总层数能被 p 整除;每卡层数不超过单卡容量。 当 micro batch 数为 m 时,气泡率为 (p-1)/(m+p-1) 。请选择气泡率最小的合法 p ;若气泡率相同,选择较小的 p 。
输入一行,必须恰好包含四个正整数:模型层数、单卡最大层数、micro batch 数、硬件卡数。每个数不超过 10^9 。 若参数数量不为四个,或任何参数不是正整数字符串,视为非法输入。
合法且存在可行并行度时,输出最优 p 和气泡率,固定保留四位小数;否则输出 `-1 -1`。
9 2 3 4
-1 -1
10 5 2
-1 -1
8 4 3 4
2 0.2500
考点:模拟
限制 1 秒 / 256MB | 标准输入输出
推荐方向:模拟
本题切入点
枚举所有满足「不超卡数、能整除层数、每卡层数不超容量」的并行度 p,逐个算气泡率 (p−1)/(m+p−1) 取最小(同值取较小 p);需先校验输入恰为四个正整数。
不涉及复杂算法,把题目描述的流程原样翻译成代码逐步执行即可。
思路框架(模拟 通法 · 非本题专属)
实现要点:结构上通常是一个外层循环包住若干 if/else 分支;只要状态定义清楚,正确率很高。
复杂度:时间 O(操作次数) | 空间 O(状态数)
该范式的通法易错点
样例 1:输入 9 2 3 4 → 输出 -1 -1
没有并行度同时满足整除、卡数和单卡容量限制。
样例 2:输入 10 5 2 → 输出 -1 -1
参数数量不是四个。
样例 3:输入 8 4 3 4 → 输出 2 0.2500
可行值为 2 和 4,气泡率随正整数并行度增大,因此选择 2。
解析由校招宝本地引擎整理(依据源站考点标签 / 人工判题标注 / 题面规模信号),非官方题解,仅供思路参考。
本题来源:2026年-华为-06月17号AI岗。