CSP 202009-1 称检测点查询 python

题目背景

2020 年 6 月 8 日，国务院联防联控机制发布《关于加快推进新冠病毒核酸检测的实施意见》，提出对“密切接触者”等八类重点人群“应检尽检”，其他人群“愿检尽检”。

问题描述

某市设有 n 个核酸检测点，编号从 1 到 n，其中 i 号检测点的位置可以表示为一个平面整数坐标 (xi,yi)。

为方便预约核酸检测，请根据市民所在位置 (X,Y)，查询距其最近的三个检测点。
多个检测点距离相同时，编号较小的视为更近。

输入格式

输入共 n+1 行。

第一行包含用空格分隔的三个整数 n、X 和 Y，表示检测点总数和市民所在位置。

第二行到第 n+1 行依次输入 n 个检测点的坐标。第 i+1 行（1≤i≤n）包含用空格分隔的两个整数 xi 和 yi，表示 i 号检测点所在位置。

输出格式

输出共三行，按距离从近到远，依次输出距离该市民最近的三个检测点编号。

样例输入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解释

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-SHYEVmDL-1637397335338)(http://118.190.20.162/RequireFile.do?fid=IwfYFZL2)]

评测用例规模与约定

全部的测试点满足，3≤n≤200，所有坐标均为整数且绝对值不超过 1000。

提示

市民到第$ i 号 检 测 点 的 距 离 号检测点的距离 号检测点的距离 D_i $可由如下公式算出：
D i 2 = ( X − x i ) 2 + ( Y − y i ) 2 D_i^2 = (X - x_i)^2 + (Y - y_i)^2 Di2​=(X−xi​)2+(Y−yi​)2

思路

这道题难度不大，但是要你对字典排序的熟悉程度就很快，我个人一直不希望遍历去找，一直在回想

然后终于得到了字典排序，通过对我们的keys排序，再得到我们的index输出就很简单了

代码

n,X,Y = map(int,input().split())

def dist(x1,y1,x2,y2):
    
    return (x1-x2)**2+(y1-y2)**2

l = {}
for i in range(n):
    x,y = map(int,input().split())
    l[i] = dist(X,Y,x,y)
# print(l)

# sorted(l.keys(), keys = lambda x:x.keys())
l = sorted(l.items(),key=lambda x:x[1])

# print(l)
# print(l[:3])
print('\n'.join(str(i[0]+1) for i in l[:3]))