求二维平面最近点对算法; 分治思想; 递归写的好精准,模仿ACM模板写的.;很多细节处理的很漂亮;

#include <stdlib.h> #include <stdio.h> #include <string.h> #include <math.h> const int N=100005; const double MAX= 10e100,eps=0.00001; //mark! struct Point { double x,y; int index; }; Point a[N],b[N],c[N]; double closest(Point *,Point *,Point *,int ,int); double dis(Point ,Point); int cmp_x(const void *,const void *); int cmp_y(const void *,const void *); void merge(Point *,Point *,int ,int ,int); inline double min(double ,double ); int main() { int n,i; double d; scanf("%d",&n); while(n) { for(i=0;i<n;++i) { scanf("%lf%lf",&a[i].x,&a[i].y); } qsort(a,n,sizeof(a[0]),cmp_x); for(i=0;i<n;++i) { a[i].index=i; } memcpy(b,a,sizeof(a[0])*n); qsort(b,n,sizeof(b[0]),cmp_y); d=closest(a,b,c,0,n-1); printf("%.2lf",d); scanf("%d",&n); } return 0; } double closest(Point *a,Point *b,Point *c,int p,int q) { if(q-p==1) { return dis(a[p],a[q]); } if(q-p==2) { double x1=dis(a[p],a[q]); double x2=dis(a[p+1],a[q]); double x3=dis(a[p],a[p+1]); double m=min(x1,x2); return min(m,x3); } int i,j,k,m=(p+q)/2; double d1,d2; for(i=p,j=p,k=m+1;i<=q;++i) { if(b[i].index<=m) { c[j++]=b[i]; } else { c[k++]=b[i]; } } d1=closest(a,c,b,p,m); d2=closest(a,c,b,m+1,q); double dm=min(d1,d2); merge(b,c,p,m,q); /* 限制横向距离 */ for(i=p,k=p;i<=q;++i) { if(fabs(b[i].x-b[m].x)<dm) { c[k++]=b[i]; } } for(i=p;i<k;++i) { for(j=i+1;j<k&&c[j].y-c[i].y<dm;++j) { double temp=dis(c[j],c[i]); if(temp<dm) { dm=temp; } } } return dm; } double dis(Point p,Point q) { double x1=p.x-q.x; double y1=p.y-q.y; return sqrt(x1*x1+y1*y1); } void merge(Point p[],Point q[],int s,int m,int t) { int i,j,k; for(i=s,j=m+1,k=s;i<=m&&j<=t;) { if(q[i].y>q[j].y) { p[k++]=q[j]; ++j; } else { p[k++]=q[i]; ++i; } } while(i<=m) { p[k++]=q[i]; ++i; } while(j<=t) { p[k++]=q[j]; ++j; } } int cmp_x(const void *p,const void *q) { double temp=((Point *)p)->x-((Point *)q)->x; if(temp>0) { return 1; } else if(fabs(temp)<eps) { return 0; } else { return -1; } } int cmp_y(const void *p,const void *q) { double temp=((Point *)p)->y-((Point *)q)->y; if(temp>0) { return 1; } else if(fabs(temp)<eps) { return 0; } else { return -1; } } inline double min(double p,double q) { return p<q? p:q; }

 

 

空间非常节约,用了一个额外的数组,每层递归完成数组的划分,回溯归来的时候再合并起来,感觉很通用的思想,以前也见过,这次记忆一下.

 

另外,为了方便划分,对a[]进行了标号,当然b[]也有标号,这样就很容易根据中点index划分数组了,省去了很多很多要考虑的麻烦.

 

另外,浮点数等于不能写==,要<eps ,因为浮点数是有很多小数点位是不精确的, 这里eps=0.00 00 1,以后就选这个数用好了.

 

这个算法也复习了一下数组的合并,数组的拆分,思想都很标准,实现也很漂亮.