· 给定三个正整数 n、m、h(均小于 100),构造如下数据并计算结果。 · 数据构造规则: · 输入特征矩阵 X 为 n×m 的全 1 矩阵。 · 三个权重矩阵 W1、W2、W3 均为 m×h 的“上三角全 1”矩阵(按行列索引在主对角线及其上方位置为 1,其余为 0;当 m≠h 时视为按行列索引的上三角扩展)。 · 令 Q=X·W1,K=X·W2,V=X·W3;计算 S=(Q·K^T)/sqrt(h)。 · softmax 按行做“归一化”:对任意行向量 r,softmax(r) 的每个元素等于该元素除以本行所有元素之和。 · Y=softmax(S)·V。 · 输出要求:求矩阵 Y 所有元素的和,四舍五入到整数后输出。
一行,三个正整数 n m h(均小于 100,且均>0)
一行,一个整数:矩阵 Y 的元素和(四舍五入后)
5 4 3
30
考点:模拟
限制 1 秒 / 256MB | 标准输入输出
推荐方向:模拟
本题切入点
纯按给定构造规则算:X 全 1、W 为上三角全 1,Q/K/V 可直接化简,S=(Q·Kᵗ)/√h 后做行 softmax(此处为「除以行和」),再乘 V 求和后四舍五入取整。
不涉及复杂算法,把题目描述的流程原样翻译成代码逐步执行即可。
思路框架(模拟 通法 · 非本题专属)
实现要点:结构上通常是一个外层循环包住若干 if/else 分支;只要状态定义清楚,正确率很高。
复杂度:时间 O(操作次数) | 空间 O(状态数)
该范式的通法易错点
样例 1:输入 5 4 3 → 输出 30
h≤m,单行和为 1+2+3=6;总和= n×6 = 5×6=30。
解析由校招宝本地引擎整理(依据源站考点标签 / 人工判题标注 / 题面规模信号),非官方题解,仅供思路参考。
本题来源:2025年秋招-华为-9月17号AI岗。