202009 CCF 称检测点查询 Python

称检测点查询

题目

题目背景
      2020 年 6 月 8 日，国务院联防联控机制发布《关于加快推进新冠病毒核酸检测的实施意见》，提出对“密切接触者”等八类重点人群“应检尽检”，其他人群“愿检尽检”。
问题描述
      某市设有 n 个核酸检测点，编号从 1 到 n，其中 i 号检测点的位置可以表示为一个平面整数坐标 (xi, yi)。
      为方便预约核酸检测，请根据市民所在位置 ，查询距其最近的三个检测点。
      多个检测点距离相同时，编号较小的视为更近。
输入格式
      输入共 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
2
3

 

样例2
输入

5 0 1
-1 0
0 0
1 0
0 2
-1 2

 

输出

2
4
1

 

解释

评测用例规模与约定
      全部的测试点满足：3<= n <= 200，所有坐标均为整数且绝对值不超过 1000。
提示
      市民到第 i 号检测点的距离 Di ，可由如下公式算出：

n, x, y = list(map(int, input().split()))
A = []
D = 0
D1 = {}
for i in range(n):
    x1, y1 = list(map(int, input().split()))
    A.append((x1, y1))
    D = pow((x1-x),2) + pow((y1-y),2)
    D1[i+1] = D
listD1 = sorted(D1.items(), key=lambda x:x[1],reverse=False)

if n < 3:
    for i in range(len(listD1)):
        print(listD1[i][0])
else:
    for i in range(3):
        print(listD1[i][0])