最接近点对问题

参考链接

#include <iostream>
#include <stdio.h>
#include <cmath>
#include <algorithm>

using namespace std;

const int MAXN=100005;

struct Point
{
    double x;
    double y;
};

Point px[MAXN],py[MAXN];

double get_dis(Point p1,Point p2)
{
    return sqrt((p1.x-p2.x)*(p1.x-p2.x)+(p1.y-p2.y)*(p1.y-p2.y));
}

bool cmpx(Point p1,Point p2)
{
    return p1.x<p2.x;
}

bool cmpy(Point p1,Point p2)
{
    return p1.y<p2.y;
}

//-----核心代码-----
double closeset(int l,int r)
{
    if(l+1==r)
        return get_dis(px[l],px[r]);
    if(l+2==r)
        return min(get_dis(px[l],px[l+1]),min(get_dis(px[l],px[r]),get_dis(px[l+1],px[r])));
    int mid=(l+r)>>1;
    //递归求解
    double mini=min(closeset(l,mid),closeset(mid+1,r));
    //把x坐标在px[mid].x-mini到px[min].x+mini中的点罗列出来。
    int cnt=0;
    for(int i=l;i<=r;i++)
    {
        if(px[i].x>=px[i].x-mini&&px[i].x<=px[i].x+mini)
            py[cnt++]=px[i];
    }
    sort(py,py+cnt,cmpy);
    for(int i=0;i<cnt;i++)
        for(int j=i+1;j<cnt;j++)
        {
            if(py[j].y-py[i].y>=mini) //因为py使按照y升序排列。
                break;
            mini=min(get_dis(py[i],py[j]),mini);
        }
    return mini;
}

int main()
{
    int n;
    while(cin>>n&&n)
    {
        for(int i=0;i<n;i++)
            cin>>px[i].x>>px[i].y;
        sort(px,px+n,cmpx);
        double mindis=closeset(0,n-1);
        printf("%f\n",mindis);
    }
    return 0;
}