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

题目描述

小红需要根据实际序列长度动态调整旋转位置编码。令缩放因子 s=max(1,T/L) ,新基频 b'=b· s^D/(D-2) 。对于第 i 对维度(从 i=0 开始),旋转角为 _i=m(b')^-2i/D 。
原向量中这一对数值为 (x_2i,x_2i+1) ,旋转后为
x'_2i=x_2i_i-x_2i+1_i,
x'_2i+1=x_2i_i+x_2i+1_i.
所有计算均使用 64 位浮点数。

输入输出

输入描述
第一行输入 T,L,D,b,m ,第二行输入 D 个实数。
保证 1≤ T,L≤10^9 , 4≤ D≤2×10^5 且 D 为偶数, 1≤ b≤10^6 , 0≤ m<T ,向量分量绝对值不超过 10^4 。
输出描述
输出旋转后的 D 个数,每个数保留 4 位小数。绝对值小于 0.00005 的结果输出为 `0.0000`。

样例共 2 组

样例 1 · 先按实际长度调整基频,再对相邻的每一对维度执行旋转。
输入
368 276 16 1450.9560044215173 209
-1.8728 1.3494 1.2030 -0.5053 -1.0223 -0.4466 1.5474 0.0967 0.0820 -2.1588 0.9543 0.0011 -0.0810 0.4860 -0.0258 0.2101
输出
-1.1873 -1.9795 0.3002 -1.2698 -1.0961 -0.2079 1.3933 -0.6801 -2.1600 0.0418 -0.2163 0.9295 -0.3734 0.3215 -0.0806 0.1957
样例 2 · 由于 T>L,需要使用动态缩放后的新基频。
输入
20 10 16 100.0 12
1.0 0.0 -1.0 0.5 0.3 -0.2 1.0 0.0 -1.0 0.5 0.3 -0.2 1.0 0.0 -1.0 0.5
输出
0.8439 -0.5366 -0.9001 0.6631 -0.2941 0.2085 -0.0147 0.9999 -1.0526 -0.3769 0.3549 -0.0634 0.9781 0.2080 -1.0476 0.3907

算法解析依据充分

考点:模拟

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

题目画像

  • 元素值域:T ≤ 1e9,L ≤ 1e9,b ≤ 1e6(注意整数类型选择,避免溢出)
  • 源站时限:2 秒(牛客口径,非本题专属门槛)

解题思路

推荐方向:模拟

本题切入点

按给定公式算 RoPE:先由 T/L 求缩放因子 s 与新基频 b′,再对每一对维度算旋转角 θ_i,按 x′_2i = x_2i·cosθ − x_2i+1·sinθ、x′_2i+1 = x_2i·sinθ + x_2i+1·cosθ 旋转,保留四位小数。

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

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

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

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

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

该范式的通法易错点

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

对照本题

  • 元素值域最大到 1e9 —— 求和 / 相乘时记得开 64 位整数。

样例解读

样例 1:输入 368 276 16 1450.9560044215173 209 / -1.8728 1.3494 1.2030 -0.5053 -1.0223 -0.4466 1.5474 0.0967 0.08 → 输出 -1.1873 -1.9795 0.3002 -1.2698 -1.0961 -0.2079 1.3933 -0.6801 -2.1600 0.0418 -0.2163 0.9295 -0.3734

先按实际长度调整基频,再对相邻的每一对维度执行旋转。

样例 2:输入 20 10 16 100.0 12 / 1.0 0.0 -1.0 0.5 0.3 -0.2 1.0 0.0 -1.0 0.5 0.3 -0.2 1.0 0.0 -1.0 0.5 → 输出 0.8439 -0.5366 -0.9001 0.6631 -0.2941 0.2085 -0.0147 0.9999 -1.0526 -0.3769 0.3549 -0.0634 0.9781 0.

由于 T>L,需要使用动态缩放后的新基频。

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

本题来源:2026年-华为-06月24号AI岗。

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