华为 · 模拟 · 算法编程题
华为 模拟 时限 1 秒 / 256 MB

题目描述

小红需要为模型训练选择流水线并行度 p 。合法的 p 必须满足: p 不超过硬件卡数;模型总层数能被 p 整除;每卡层数不超过单卡容量。
当 micro batch 数为 m 时,气泡率为 (p-1)/(m+p-1) 。请选择气泡率最小的合法 p ;若气泡率相同,选择较小的 p 。

输入输出

输入描述
输入一行,必须恰好包含四个正整数:模型层数、单卡最大层数、micro batch 数、硬件卡数。每个数不超过 10^9 。
若参数数量不为四个,或任何参数不是正整数字符串,视为非法输入。
输出描述
合法且存在可行并行度时,输出最优 p 和气泡率,固定保留四位小数;否则输出 `-1 -1`。

样例共 3 组

样例 1 · 没有并行度同时满足整除、卡数和单卡容量限制。
输入
9 2 3 4
输出
-1 -1
样例 2 · 参数数量不是四个。
输入
10 5 2
输出
-1 -1
样例 3 · 可行值为 2 和 4,气泡率随正整数并行度增大,因此选择 2。
输入
8 4 3 4
输出
2 0.2500

算法解析依据充分

考点:模拟

限制 1 秒 / 256MB | 标准输入输出

题目画像

  • 源站时限:1 秒(牛客口径,非本题专属门槛)

解题思路

推荐方向:模拟

本题切入点

枚举所有满足「不超卡数、能整除层数、每卡层数不超容量」的并行度 p,逐个算气泡率 (p−1)/(m+p−1) 取最小(同值取较小 p);需先校验输入恰为四个正整数。

不涉及复杂算法,把题目描述的流程原样翻译成代码逐步执行即可。

思路框架(模拟 通法 · 非本题专属)

  1. 用变量记录题目要求的「状态」(当前值、剩余数量、当前位置等)。
  2. 按题面给出的顺序,把每一步操作写成一段代码,逐条执行。
  3. 每一步执行后更新状态,并在题目要求的位置输出或累计答案。
  4. 注意循环的边界:执行多少次、何时终止、是否能终止。

实现要点:结构上通常是一个外层循环包住若干 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岗。

‹ 上一题 全部编程题 下一题 ›
编程算法题为只读内容:无需作答,直接看题与解析 · 本站不提供在线判题 · 解析由校招宝本地引擎整理,非官方题解