最近点对,采用分治方法。过程:
1对原数组依据x左标从小到大排序。
2二分数组,左边求出最小值,右边求出最小值,我们求最小的。
3找出对于左右两边的可能小于当前最小值的最近点对,更新最小值。
这题目需要区分一下点,让我们求的是闪兵到任意一个核电站的最短距离,加一个标志就可以了。
代码:
#include <iostream> #include <stdio.h> #include <iomanip> #include <algorithm> #include <cmath> using namespace std; const int N=200000+1; const double INF=1e100; struct Point{ double x,y; bool flag; }; Point m[N+1]; int tmp[N+1]; double dis(Point p1,Point p2) { return sqrt((p1.x-p2.x)*(p1.x-p2.x)+(p1.y-p2.y)*(p1.y-p2.y)); } double minL(double l1,double l2) { return l1<=l2?l1:l2; } bool cmpy(int a,int b) { return m[a].y<m[b].y; } bool cmpx(Point a,Point b) { return a.x<b.x; } double getMinLen(Point *s,int left,int right) { double rs=INF; if(left == right) return rs; if(left+1 == right) { if(s[left].flag == s[right].flag) return rs; return dis(s[left],s[right]); } int mid=(left+right)>>1; rs=getMinLen(s,left,mid); rs=minL(rs,getMinLen(s,mid+1,right));//rs现在是两边中最点值了 int i,j,num=0; for(i=left;i<=right;++i) { if(fabs(s[i].x-s[mid].x)<=rs)//过滤掉不可能是最近点对的点 tmp[num++]=i; } sort(tmp,tmp+num,cmpy);//按y坐标排序,减少比较次数 double d=INF; for(i=0;i<num;++i) { for(j=i+1;j<num;++j) { if(fabs(s[tmp[i]].y-s[tmp[j]].y)>=rs)//i,j两点的y坐标已经大于rs,不需要在遍历了 break; if(s[tmp[i]].flag!=s[tmp[j]].flag && (d=dis(s[tmp[i]],s[tmp[j]]))<rs) rs=d; } } // cout<<left<<" "<<right<<" "<<rs<<endl; return rs; } int main() { int t,n,i; scanf("%d",&t); while(t--) { scanf("%d",&n); for(i=0;i<n;++i) { scanf("%lf%lf",&m[i].x,&m[i].y); m[i].flag=0; } for(i=0;i<n;++i) { scanf("%lf%lf",&m[i+n].x,&m[i+n].y); m[i+n].flag=1; } n<<=1; sort(m,m+n,cmpx); double ans=getMinLen(m,0,n-1); cout<<setiosflags(ios::fixed)<<setprecision(3)<<ans<<endl; } return 0; }