最近餐馆

1:最近餐馆

总时间限制: 

5000ms 

内存限制: 

98304kB

描述

每到饭点，就又到了一日几度的小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 21 11 33 422 322 31

样例输出

the closest 2 points are:1 33 4the closest 1 points are:1 3

#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<iomanip>
#include<queue>
#include<stack>
#include<vector>
#include<set>
#include<map>
using namespace std;
struct Node
{
	int Max[5];
	int Min[5];
	int Num[5];
	int l,r;
}KD[5005];
int root,n,k,t,cur,m;
int p[5];
int an[15];
priority_queue<pair<int,int> >q;
bool cmp(const Node&x,const Node&y)
{
	return x.Num[cur]<y.Num[cur];
}
void Update(int root)
{
	int l=KD[root].l,r=KD[root].r;
	for(int i=0;i<k;++i)
	{
		int Min=min(KD[l].Min[i],KD[r].Min[i]);
		int Max=max(KD[l].Max[i],KD[r].Max[i]);
		if(KD[root].Min[i]>Min) KD[root].Min[i]=Min;
		if(KD[root].Max[i]<Max) KD[root].Max[i]=Max;
	}
}

int BuildKdtree(int l,int r,int level)
{
	cur=level;
	int mid=(l+r)>>1;
	nth_element(KD+l,KD+mid,KD+r+1,cmp);
	if(mid>l)KD[mid].l=BuildKdtree(l,mid-1,(level+1)%k);
	else KD[mid].l=0;
	if(r>mid)KD[mid].r=BuildKdtree(mid+1,r,(level+1)%k);
	else KD[mid].r=0;
	Update(mid);
	return mid;
}
int Dis(int x)
{
	int ans=0;
	for(int i=0;i<k;++i)
	{
		ans+=(KD[x].Num[i]-p[i])*(KD[x].Num[i]-p[i]);
	}
	return ans;
}
int GetDis(int x)
{
	int ans=0;
	for(int i=0;i<k;++i)
	{
		if(p[i]<KD[x].Min[i])
		{
			ans+=(p[i]-KD[x].Min[i])*(p[i]-KD[x].Min[i]);
		}
		if(p[i]>KD[x].Max[i])
		{
			ans+=(p[i]-KD[x].Max[i])*(p[i]-KD[x].Max[i]);
		}
	}
	return ans;
}
void Dfs(int root)
{
	if(root==0)return;
	int d=Dis(root),dl=GetDis(KD[root].l),dr=GetDis(KD[root].r);
	if(d<q.top().first)
	{
		q.pop();
		q.push(make_pair(d,root));
	}
	if(dl<dr)
	{
		if(dl<q.top().first)
			Dfs(KD[root].l);
		if(dr<q.top().first)
			Dfs(KD[root].r);
	}
	else
	{
		if(dr<q.top().first)
			Dfs(KD[root].r);
		if(dl<q.top().first)
			Dfs(KD[root].l);
	}
}
int main()
{
	while(cin>>n>>k)
	{
		for(int i=0;i<k;++i)
		{
			KD[0].Max[i]=-0x7fffffff/2;
			KD[0].Min[i]=0x7fffffff/2;
		}
		for(int i=1;i<=n;++i)
			for(int j=0;j<k;++j)
			{
				cin>>KD[i].Num[j];
				KD[i].Min[j]=KD[i].Max[j]=KD[i].Num[j];
			}
			

		root=BuildKdtree(1,n,0);
		cin>>t;
		while(t--)
		{
			for(int i=0;i<k;++i)
			{
				cin>>p[i];
			}
			cin>>m;
			cout<<"the closest "<<m<<" points are:"<<endl;
			while(!q.empty())q.pop();
			for(int i=1;i<=m;++i)
			{
				q.push(make_pair(0x7fffffff/2,0));
			}
			Dfs(root);
			for(int i=1;i<=m;++i)
			{
				an[i]=q.top().second;
				q.pop();
			}
			for(int i=m;i>=1;--i)
			{
				for(int j=0;j<k;++j)
				{
					cout<<KD[an[i]].Num[j]<<" ";
				}
				cout<<endl;
			}
		}
	}
	return 0;
}