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