PAT乙级1082 射击比赛//寻求更好的方法

本题目给出的射击比赛的规则非常简单，谁打的弹洞距离靶心最近，谁就是冠军；谁差得最远，谁就是菜鸟。本题给出一系列弹洞的平面坐标(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

思路：这题很简单，用一个结构体保存选手id以及成绩，列入vector容器中，并用sort按照成绩排列（注意距离是越短的成绩越好）。但是我的做法时间复杂度达到了O(nlogn)

#include <iostream>
#include <algorithm>
#include <vector>
#include <cmath>
#include <string>
using namespace std;
struct data
{
	string id;
	int x;
	int y;
	int final;
};
bool cmp(data a, data b) { return a.final < b.final; }
int main()
{
	int n;
	cin >> n;
	vector<data> vec(n);
	for (int i = 0; i < n; i++)
	{
		cin >> vec[i].id >> vec[i].x >> vec[i].y;
		vec[i].final = sqrt((double)vec[i].x*vec[i].x + vec[i].y*vec[i].y);
	}
	sort(vec.begin(), vec.end(), cmp);
	cout << vec[0].id << ' ' << vec[vec.size() - 1].id;
	return 0;
}

但是可以一边输入一边比较，将时间复杂度优化到O(n)

#include <iostream>
#include <cmath>
using namespace std;
int main()
{
	int n, id, x, y, max_id, max_sco = -1, min_id, min_sco = INT_MAX;
	cin >> n;
	for (int i = 0; i < n; i++)
	{
		cin >> id >> x >> y;
		int dis = x * x + y * y;
		if (dis > max_sco)
		{
			max_sco = dis;
			max_id = id;
		}
		if (dis < min_sco)
		{
			min_sco = dis;
			min_id = id;
		}
	}
	printf("%04d %04d", min_id, max_id);
	return 0;
}