PTA 1082 射击比赛(Python3)

本题目给出的射击比赛的规则非常简单，谁打的弹洞距离靶心最近，谁就是冠军；谁差得最远，谁就是菜鸟。本题给出一系列弹洞的平面坐标(x,y)，请你编写程序找出冠军和菜鸟。我们假设靶心在原点(0,0)。

输入格式：

输入在第一行中给出一个正整数 N（≤ 10 000）。随后 N 行，每行按下列格式给出：

ID x y

其中 ID 是运动员的编号（由 4 位数字组成）；x 和 y 是其打出的弹洞的平面坐标(x,y)，均为整数，且 0 ≤ |x|, |y| ≤ 100。题目保证每个运动员的编号不重复，且每人只打 1 枪。

输出格式：

输出冠军和菜鸟的编号，中间空 1 格。题目保证他们是唯一的。

输入样例：

3
0001 5 7
1020 -1 3
0233 0 -1

输出样例：

0233 0001

提交结果：

代码：

import math
 
num = int(input())
l = []
ll = []
l3 = []
for i in range(num):
    data = input().split()
    juli = math.sqrt(pow(int(data[1]), 2) + pow(int(data[2]), 2))
    l.append(juli)
    ll.append([data[0], juli])
max1 = max(l)
min1 = min(l)
for i in ll:
    if i[1] == min1:
        l3.append(i[0])
        break
for i in ll:
    if i[1] == max1:
        l3.append(i[0])
        break
print(' '.join(l3))