题目背景
2020 年 6 月 8 日,国务院联防联控机制发布《关于加快推进新冠病毒核酸检测的实施意见》,提出对“密切接触者”等八类重点人群“应检尽检”,其他人群“愿检尽检”。
问题描述
某市设有 nnn 个核酸检测点,编号从 111 到 nnn,其中 iii 号检测点的位置可以表示为一个平面整数坐标 (xi,yi)(x_i, y_i)(xi,yi)。
为方便预约核酸检测,请根据市民所在位置 (X,Y)(X, Y)(X,Y),查询距其最近的三个检测点。
多个检测点距离相同时,编号较小的视为更近。
输入格式
输入共 n+1n+1n+1 行。
第一行包含用空格分隔的三个整数 nnn、XXX 和 YYY,表示检测点总数和市民所在位置。
第二行到第 n+1n+1n+1 行依次输入 nnn 个检测点的坐标。第 i+1i+1i+1 行(1≤i≤n1 \le i \le n1≤i≤n)包含用空格分隔的两个整数 xix_ixi 和 yiy_iyi,表示 iii 号检测点所在位置。
输出格式
输出共三行,按距离从近到远,依次输出距离该市民最近的三个检测点编号。
样例输入1
3 2 2
2 2
2 3
2 4
样例输出1
1
2
3
样例输入2
5 0 1
-1 0
0 0
1 0
0 2
-1 2
样例输出2
2
4
1
样例2解释
评测用例规模与约定
全部的测试点满足,3≤n≤2003 \le n \le 2003≤n≤200,所有坐标均为整数且绝对值不超过 100010001000。
提示
市民到第 iii 号检测点的距离 DiD_iDi 可由如下公式算出:
Di2=(X−xi)2+(Y−yi)2D_i^2 = (X - x_i)^2 + (Y - y_i)^2Di2=(X−xi)2+(Y−yi)2
解法
import math
n, X, Y = list(map(int, input().split()))
# 存储核酸监测点编号及其坐标
jcd = {}
for i in range(n):
x, y = list(map(int, input().split()))
jcd[i + 1] = [x, y]
# 存储核酸检测点编号及市民到该点的距离
dis = {}
for k, v in jcd.items():
d = math.sqrt((X - v[0])**2 + (Y - v[1])**2)
dis[k] = d
dis = sorted(dis.items(), key=lambda x: x[1]) # 按v值从小到大排序
for idx, d in enumerate(dis):
print(d[0])
if idx == 2:
break