poj:4091:The Closest M Points
poj:4091:The Closest M Points
题目
每到饭点,就又到了一日几度的小L纠结去哪吃饭的时候了。由于有太多太多好吃的地方能够去吃,而小L又比較懒不想走太远,所以小L会先找到距离他近期的M家餐馆然后再做筛选。
小L如今所在的位置和每家餐馆的位置用同一笛卡尔坐标系中的点表示,而点与点之间的距离为欧几里得距离,对于点p = (p1, p2,..., pn)和点q = (q1,q2,..., qn),两者的距离定义例如以下
现给出在K维空间中小L所处的位置的坐标以及n个餐馆的位置,请帮助小L完毕他的需求。
接下来n行,每行包括K个数,表示每一个餐馆的坐标。
接下来1行。包括一个数t,1≤t≤10000。表示小L询问的数目。
每次询问包括两行。
第1行包括K个数,表示小L所在的坐标。第2行包括一个数M,1≤M≤10。
全部坐标值不会超过10000。
输入数据包括多组数据。请逐个处理直到文件结束。
第1行输出:”the closest M points are:”,当中M在输入中给出。
接下来M行输出距离近期的M家餐馆的坐标,依照由近及远的顺序输出。
输出数据保证答案唯一。保证从小L的位置到近期的M+1家餐馆位置各不同样。这说明例如以下输入数据:
2 2
1 1
3 3
1
2 2
1
不会存在。
3 2 1 1 1 3 3 4 2 2 3 2 2 3 1
the closest 2 points are: 1 3 3 4 the closest 1 points are: 1 3
解题方案,这个题目比較简单
我们转换问题这是求出k最大值或者最小值问题
暴力法是先排序再求出前m个最小值
快捷法是建立m规模的小顶堆。然后去全部数据去更新一边,最后全部的堆的元素就是结果
代码(暴力法代码)
#include <iostream> #include <fstream> #include <list> #include <math.h> #include <algorithm> using namespace std; typedef pair<double,int> distance_restaurant; void read_data(); // data 为餐馆地址表 list<distance_restaurant> main_solution(int ** data,int n,int k,int *dizhi); int main() { read_data(); system("pause"); return 0; } void read_data() { ifstream reader; reader.open("data.txt"); int n,k; reader>>n; reader>>k; int ** data = new int *[n]; for( int i=0; i<n; i++ ) { data[i] = new int[k]; for(int j=0;j<k;j++) reader>>data[i][j]; } int t; reader>>t; while( t>0 ) { t-- ; int * dizhi = new int[k] ; for( int i=0;i<k;i++ ) reader>>dizhi[i] ; int m; reader>>m; cout<<"the closest "<<m<<" points are:"<<endl; list<distance_restaurant> result = main_solution(data,n,k,dizhi); for( list<distance_restaurant>::iterator it = result.begin();m>0 && it != result.end() ;m-- ,it++) { for(int j=0;j<k;j++) { cout<<data[it->second][j]<<" "; } cout<<endl; } } } // data 为餐馆地址表 list<distance_restaurant> main_solution(int ** data,int n,int k,int *dizhi) { list<distance_restaurant> result; distance_restaurant elem; for( int i=0;i<n;i++) { elem.second = i; elem.first = 0; for(int j=0;j<k;j++) { elem.first += (data[i][j]-dizhi[j]) * (data[i][j]-dizhi[j]); } elem.first = sqrt(elem.first); result.push_back(elem); } stable_sort(result.begin(),result.end()); return result; }