最近餐馆

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