题目大意
给定一个点(x,y)及其他n个坐标,求离(x,y)最近的三个点的坐标的编号,若距离相等则取编号小的坐标。
解法1
- 排序
- 简单容器处理
解题思路
- 借助multimap来保存<距离,编号>的键值对,借助它允许相同键及按键排序(相同键按值自动排序)的特性来存储结果。
- 使用循环不断读取题意坐标,对坐标不断做距离计算,插入到容器中,若容器大于3个则删除最后一个键值对,循环完毕后,容器内的键值对即是题意所求的答案。
AC代码 时间复杂度O(n) 空间复杂度O(1)
#include<iostream>
#include<map>
#include<cmath>
using namespace std;
int main(){
int n,X,Y;
multimap<int,int> q;//multimap<距离,编号>存最近的三个点的距离及编号
cin>>n>>X>>Y;//题目输入
for(int i=0,xi,yi;i<n;++i){//遍历输入的所有监测点坐标
cin>>xi>>yi;
int absX = abs(X-xi);
int absY = abs(Y-yi);
int distance = absX*absX + absY*absY;//这里避免开平方所取的小数造成判断错误
q.insert(pair<int,int>(distance,i+1));
if(q.size()>3){
q.erase(--q.end());//删掉最后一个元素
}
}
//输出容器内的结果
for(multimap<int,int>::iterator it = q.begin();it!=q.end();++it){
cout<<it->second<<endl;
}
return 0;
}
解法2
#include<cmath>
#include<iostream>
#include<algorithm>
using namespace std;
struct node {
int id;
int distance;
};
bool cmp(node n1, node n2) {
if(n1.distance==n2.distance)
return n1.id<n2.id;
return n1.distance<n2.distance;
}
int main() {
int n;
cin>>n;
int X,Y;
cin>>X>>Y;
node point[n];
for(int i=0; i<n; i++) {
int pointx;
int pointy;
point[i].id=i+1;
cin>>pointx>>pointy;
point[i].distance=(pointx-X)*(pointx-X) + (pointy-Y)*(pointy-Y);
}
sort(point,point+n,cmp);
for(int i=0; i<3; i++)
cout<<point[i].id<<endl;
return 0;
}
sort函数的注意事项
1.头文件
#include <algorithm>
2.命名空间
using namespace std;
3.自定义排序函数
bool cmp(a, b)//从大到小排序
{
return a>b;
}
4.使用
sort(point, point+n, cmp);//n是point数组的长度