#AEAI02006. 最近成功抓取模板

最近成功抓取模板

题目描述

具身智能部的抓取库保存了许多历史成功案例,每个案例都有场景特征和对应的抓取模板标签。面对新的观测,系统希望先找出最相似的一次历史抓取,复用它的模板作为基线策略。

给定各维特征的距离权重、历史样本和查询特征,请计算查询与每个历史样本之间的加权平方距离。选择距离最小的样本,输出它的编号、模板标签和对应距离。

距离不需要开平方;若多个历史样本距离完全相同,选择输入顺序中编号更小的样本。

输入格式

  • 第一行包含 n d。
  • 第二行包含 d 个距离权重。
  • 随后 n 行每行包含 d 个历史特征和模板标签。
  • 最后一行包含 d 个查询特征。

输出格式

  • 第一行输出最近历史样本编号。
  • 第二行输出模板标签。
  • 第三行输出最小加权平方距离。

数据范围与约束

  • 1 <= n <= 10000
  • 1 <= d <= 20
  • 权重非负

判定规则

使用标准输出比较。浮点结果按题面保留 6 位或 10 位小数;阈值等号、无可行分支和 Tie-break 按下列规则处理。

  • 距离为 sum(weight_j*(x_j-q_j)^2)。
  • 距离相同取样本编号更小。
  • 不对距离开平方。

样例

输入样例 1

3 2
1 1
0 0 5
2 0 7
0 3 9
1 0

输出样例 1

1
5
1.000000

输入样例 2

2 1
1
0 8
2 9
1

输出样例 2

1
8
1.000000

样例解释

样例 1 中查询点到编号 1 的距离最小,因此复用标签 5。样例 2 中两个历史样本距离相同,按编号更小选择编号 1