美团 · 模拟 · 算法编程题
美团 模拟 时限 3 秒 / 256 MB

题目描述

给定一批训练样本与若干测试样本,请你手写实现主成分分析 (PCA) 并仅保留第一主成分来压缩-重建数据,最后输出每个测试样本在重建后的均方误差 (MSE)。
1. 输入读取
• train - 二维列表,每行是一个 m 维数值特征向量
• test - 二维列表,维度同上
2. 去均值 (mean-center)
X_c = X - , =mean(X_train)
3. 协方差矩阵 (总体方差,ddof=0)
= 1n X_c^X_c
4. 求第一主成分
• 用 numpy.linalg.eigh 得到全部特征对 (_i,v_i)
• 按特征值从大到小选取第一主成分 v_max
• 方向标准化规则 —— 若 v_max 首个非零分量为负,则整体乘以 -1;这样方向唯一
5. 投影-重建
z = (x-)^ vmax, x = + zvmax
6. 输出
• 对每个测试样本计算
MSE(x) = 1m Σ_j=1^m (x_j- x_j)^2
• 结果保留两位小数,使用字符串形式
• 所有测试样本的误差按输入顺序组成 JSON 数组一行输出

输入输出

输入描述
标准输入仅一行,为如下 JSON 对象:
{
"train": [[...], [...], ...],
"test": [[...], [...], ...]
}
其中
• train 长度 n≥2 ,每行长度 m≥2
• test 任意条数,维度同 train
• 所有值为整数或浮点数,无额外空行
输出描述
标准输出仅一行 —— 测试集中 每个样本 MSE 的字符串形式(两位小数),用 JSON 数组包裹。

样例共 1 组

样例 1
输入
{"train": [[0,0],[0,1],[1,0],[1,1]], "test": [[0.5,0.5],[1.5,1.5]]}
输出
["0.00", "0.50"]

算法解析依据充分

考点:模拟

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

题目画像

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

解题思路

推荐方向:模拟

本题切入点

按给定公式手写 PCA:去均值 → 协方差(ddof=0)→ eigh 取最大特征值向量 → 按「首个非零分量非负」统一方向 → 投影重建 → 逐样本 MSE,关键是严格照公式实现并注意方向唯一化。

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

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

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

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

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

该范式的通法易错点

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

样例

样例 1

  • 输入:{"train": [[0,0],[0,1],[1,0],[1,1]], "test": [[0.5,0.5],[1.5,1.5]]}
  • 输出:["0.00", "0.50"]

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

本题来源:2025年秋招-美团-算法策略端-第一批笔试。

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