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完成他的需求。

输入

第1行包含两个整数n和K，1≤n≤5000，1≤K≤5。
接下来n行，每行包含K个数，表示每个餐馆的坐标。
接下来1行，包含一个数t，1≤t≤10000，表示小L询问的数目。
每次询问包括两行。第1行包含K个数，表示小L所在的坐标。第2行包含一个数M，1≤M≤10。
所有坐标值不会超过10000。
输入数据包含多组数据，请逐个处理直到文件结束。

输出

对于每一个询问，输出m+1行：
第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;
}