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

题目描述

为了更快追踪突发热点,我们仅在“查询时刻 t 之前的最近 K 篇文档”内计算 TF‑IDF,并以加权余弦相似度挑选最相关的文档。
窗口内越新的文档权重越高(从旧到新第 j 篇的权重为 (K−j+1)/K)。
给定按时间递增的文档序列和若干查询(每条查询含时间点 t 与查询短语 q),请在窗口中找出与 q 的加权余弦相似度最高且相似度≥0.6 的文档编号;若存在并列最高,返回窗口中最早的那一篇;若无满足阈值的文档,输出 -1。
·
词向量用 TF‑IDF:TF 为词频;IDF 采用平滑公式 IDF(x)=log((N+1)/(df(x)+1))+1,其中 N 为窗口文档数,df(x) 为窗口内包含词 x 的文档数。
·
余弦相似度采用 q 与每个文档向量的点积除以范数乘积;文档向量还需乘以其时间权重。
·
文档与查询均以空格分词、统一小写,不做额外清洗。为避免早期窗口不足的问题,测试均保证 t ≥ K−1。

输入输出

输入描述
第一行:文档总数 N
接下来 N 行:按时间从 0 到 N−1 的文档内容(小写,空格分词)
下一行:窗口大小 K
下一行:查询总数 P
接下来 P 行:每行“t 空格 q”表示在时间点 t 的查询 q
输出描述
输出 P 个数字,空格分隔;每个数字是对应查询的文档编号或 -1

样例共 1 组

样例 1 · 对 t=4,窗口为文档[2,3,4]。q="finance market" 与 2、4 的原始余弦相似度相同且约为 0.605≥0.6;时间权重越新越大(2:1/3, 3:2/3, 4:1),加权后 4 更高,返回 4。 对 t=5,窗口为[2,3,4]。q="ai model" 仅与文档3匹配(含 ai、model),原始余弦≈0.707≥0.6,返回 3。 对 t=3,窗口为[1,2,3]。q="travel guide" 窗口内均无重合词,余弦=0<0.6,返回 -1。
输入
5
breaking news finance market
sports football world cup
finance stock market rises
tech ai model training
finance market crash report
3
3
4 finance market
5 ai model
3 travel guide
输出
4 3 -1

算法解析依据充分

考点:模拟

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

题目画像

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

解题思路

推荐方向:模拟

本题切入点

按定义实现检索计算:先用时间权重 (K−j+1)/K 维护窗口,再算窗口内 TF-IDF(平滑 IDF)与加权余弦相似度,取相似度 ≥0.6 中最优且编号最小者。

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

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

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

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

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

该范式的通法易错点

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

样例解读

样例 1:输入 5 / breaking news finance market / sports football world cup / finance stock market rises / tech ai → 输出 4 3 -1

对 t=4,窗口为文档[2,3,4]。q="finance market" 与 2、4 的原始余弦相似度相同且约为 0.605≥0.6;时间权重越新越大(2:1/3, 3:2/3, 4:1),加权后 4 更高,返回 4。

对 t=5,窗口为[2,3,4]。q="ai model" 仅与文档3匹配(含 ai、model),原始余弦≈0.707≥0.6,返回 3。

对 t=3,窗口为[1,2,3]。q="travel guide" 窗口内均无重合词,余弦=0<0.6,返回 -1。

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

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

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