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

题目描述

·
任务: 给定一行数据,描述一个长度为 seq_len、每步维度为 x_dim 的输入序列。使用一个固定参数的 LSTM 对序列做前向计算,并输出每个时间步隐藏向量的首元素 h_t[0]。
·
模型设定:
·
记忆单元个数 m=5。
·
初始状态 s0 为全1向量,h0 为全0向量。
·
四门权重与偏置全为0,因此每步都有 i=f=o=0.5、g=0,递推得到 s_t=0.5^t·s0,h_t=0.5·tanh(s_t)。故 h_t[0]=0.5·tanh(0.5^t)。
·
说明: 输出与具体输入值无关(由固定参数决定),仅与 seq_len 有关;这样仍符合“按所给 LSTM 前向形式计算并取首元素”的题意。

输入输出

输入描述
· 一行: seq_len x_dim 后接 seq_len·x_dim 个浮点数(按行平铺)。
输出描述
· 一行: 依次输出 t=1..seq_len 的 h_t[0],用空格分隔,四舍五入到小数点后三位,去掉多余尾零;数值为0统一输出0.0。

样例共 1 组

样例 1 · 因 s0≠0,h1[0]=0.5·tanh(0.5)=0.231,h2[0]=0.5·tanh(0.25)=0.122,h3[0]=0.5·tanh(0.125)=0.062(四舍五入到小数点后三位)。
输入
3 4 1 2 3 4 5 6 7 8 9 10 11 12
输出
0.231 0.122 0.062

算法解析依据充分

考点:模拟

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

题目画像

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

解题思路

推荐方向:模拟

本题切入点

题目已把 LSTM 递推化简成闭式:s_t = 0.5ᵗ·s0、h_t = 0.5·tanh(s_t),因此只需按 seq_len 迭代输出 h_t[0],格式化到三位小数。

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

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

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

实现要点:结构上通常是一个外层循环包住若干 if/else 分支;只要状态定义清楚,正确率很高。

复杂度:时间 O(操作次数) | 空间 O(状态数)

该范式的通法易错点

  • 终止条件写错导致死循环或漏做最后一次操作。
  • 状态更新顺序颠倒(先改了下标又用旧下标)。
  • 题目里「最多 / 恰好 / 至少」的语义差别没区分。

样例解读

样例 1:输入 3 4 1 2 3 4 5 6 7 8 9 10 11 12 → 输出 0.231 0.122 0.062

因 s0≠0,h1[0]=0.5·tanh(0.5)=0.231,h2[0]=0.5·tanh(0.25)=0.122,h3[0]=0.5·tanh(0.125)=0.062(四舍五入到小数点后三位)。

解析由校招宝本地引擎整理(依据源站考点标签 / 人工判题标注 / 题面规模信号),非官方题解,仅供思路参考。

本题来源:2025年秋招-华为-10月10号AI岗。

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