CSP 202009-1 称检测点查询 python

输入格式

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

评测用例规模与约定

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

提示

市民到第$ i 号 检 测 点 的 距 离 号检测点的距离号检测点的距离 D_i $可由如下公式算出：

思路

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

然后终于得到了字典排序，通过对我们的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]))