CSP202009-1 称检测点查询 (Python)

题目

试题编号：	202009-1
试题名称：	称检测点查询
时间限制：	1.0s
内存限制：	256.0MB

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

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

D~i~^2^=(X - x~i~)^2^+(Y - y~i~)^2^

程序代码

# 用于距离列表排序时返回距离值
def take_2(List):
    return List[1]

# 输入检测点个数及市民位置并转换类型
n,X,Y = input().split()
n = int(n)
X = int(X)
Y = int(Y)

# 检测点距离列表
distance = []

# 循环输入监测点坐标并计算和记录距离
for i in range(n):
    x,y = input().split()
    x = int(x)
    y = int(y)
    # 此时直接比较D^2即可
    d = (X - x)**2 + (Y - y)**2
    distance.append([i + 1,d])

# 按照距离顺序排序
distance.sort(key=take_2)

# 输出排序后前三个检测点的编号
for i in range(3):
    print(distance[i][0])