[分治] poj3714:[算法竞赛进阶指南]Raid (0x08「总结与练习」)

版权声明:本文为QQQQqtt原创,但是我猜没有人转载oAo https://blog.csdn.net/qq_36038511/article/details/80970417

poj3714 Raid:http://poj.org/problem?id=3714

平面最近点对的裸题了吧
之前没有写过
按x坐标作为第一关键字排序后用分治解决
合并子问题是只用考虑跨越了左右两个子问题的部分
对于当前已有的最小值确定出mid周围一个正方形的区域 确定这些点的最小距离即可 (据说易证不超过8个?)
按y坐标排序后乱搞
复杂度是 n(logn)^2 用归并排序可以少一个log?

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
struct node
{
    double x,y; int z;
}p[210000],t[210000],tt[210000];
bool cmp1(node x,node y) {return x.x<y.x;}
bool cmp2(node x,node y) {return x.y<y.y;}
double mymin(double x,double y){if (x>y) return y; else return x;}
double dis(node x,node y)
{
    if (x.z==y.z) return (1<<20);
    return sqrt((x.x-y.x)*(x.x-y.x)+(x.y-y.y)*(x.y-y.y));
}
double fenzhi(int l,int r)
{
    if (l==r) return (1<<20);
    //归并排序 以 y 为第二关键字排序 
    int mid=(l+r)/2; double minn=mymin(fenzhi(l,mid),fenzhi(mid+1,r));
    int i=l,j=mid+1,pp=l;
    while (i<=mid&&j<=r)
    {
        if (p[i].y<=p[j].y) t[pp++]=p[i++];
        else t[pp++]=p[j++];
    }
    while (i<=mid) t[pp++]=p[i++];
    while (j<=r) t[pp++]=p[j++];
    for (int k=l;k<=r;k++) p[k]=t[k];

    //确定出mid在mn正方形范围内的点 
    int len=0; tt[++len]=p[mid];
    for (int i=l;i<=r;i++) if (fabs(p[i].x-p[mid].x)<=minn) tt[++len]=p[i];

    //直接乱搞更新 
    for (int i=1;i<=len;i++)
    {
        for (int j=i+1;j<=len;j++)
        {
            if (tt[i].y-tt[j].y>minn) break;
            minn=mymin(dis(tt[i],tt[j]),minn);
        }
    }
    return minn;
}
int main()
{
    int u;
    scanf("%d",&u);
    while (u--)
    {
        memset(t,0,sizeof(t));
        int n;
        scanf("%d",&n);
        for (int i=1;i<=n;i++)
        {
            scanf("%lf%lf",&p[i].x,&p[i].y);
            p[i].z=1;
        }
        for (int i=1;i<=n;i++)
        {
            scanf("%lf%lf",&p[i+n].x,&p[i+n].y);
            p[i+n].z=2;
        }
        sort(p+1,p+1+2*n,cmp1); //先以 x 为第一关键字排序 
        printf("%.3lf\n",fenzhi(1,n*2));
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_36038511/article/details/80970417
今日推荐