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;
}