最近点对的问题。题意给定两个点集,求两个点集中的点的最小距离。
大体思路:找到对两堆点集进行标记。然后按照x排序。在求最近点对的时候,找到距离中间点最近点加入数组b中。然后求数组b中且不是同一个点集中的最小的距离。最终返回最小距离。
代码如下:
#include<stdio.h>
#include<math.h>
#include<iostream>
#include<algorithm>
using namespace std;
#define N 200003
#define INF 1e18
struct point
{
double x;
double y;
bool flag;
}p[N],b[N];
bool cmpx(point a,point b)
{
return a.x<b.x;
}
bool cmpy(point a,point b)
{
return a.y<b.y;
}
double dis(point a,point b)
{
return sqrt((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y));
}
double min(double a,double b)
{
return a>b?b:a;
}
double mindis(int l,int r)
{
int mid=(l+r)/2,i;
double res=INF;
if(l>=r)return res;
for(i=mid;i>=l&&(p[i].x==p[mid].x);i--);
res=mindis(l,i);
for(i=mid;i<=r&&(p[i].x==p[mid].x);i++);
res=min(res,mindis(i,r));
int k=0;
for(int i=mid;i>=l&&(p[mid].x-p[i].x<=res);i--)
{
b[k++]=p[i];
}
for(int i=mid+1;i<=r&&(p[i].x-p[mid].x<=res);i++)
{
b[k++]=p[i];
}
sort(b,b+k,cmpy);
for(int i=0;i<k;i++)
{
for(int j=i+1;j<k;j++)
{
if(b[j].y-b[i].y<res&&b[j].flag!=b[i].flag)
{
res=min(res,dis(b[i],b[j]));
}
}
}
return res;
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
int n;
scanf("%d",&n);
for(int i=0;i<n+n;i++)
{
scanf("%lf%lf",&p[i].x,&p[i].y);
if(i<n)p[i].flag=false;
else p[i].flag=true;
}
n<<=1;
sort(p,p+n,cmpx);
printf("%.3lf\n",mindis(0,n-1));
}
return 0;
}