最接近点对(分治)

代码如下:

#include<bits/stdc++.h>
using namespace std;
const int MAXN = 10010;

//定义结构体 
struct Point {
    double x, y;
} p[MAXN];

// 按照 x 坐标排序
bool cmp(Point a, Point b) {
    return a.x < b.x;
}

// 计算距离
double dist(Point a, Point b) {
    double dx = a.x - b.x, dy = a.y - b.y;
    return sqrt(dx * dx + dy * dy);
}

//合并 
double merge(int l, int r, double d) {                                
    double midx = (p[l].x + p[r].x) / 2.0;//计算两侧的中心点x坐标的平均值 
    double ans = d;
    int k = 0;        //标记tmp[]数组下标 
    Point tmp[MAXN];//临时数组tmp[]
    
    //(x)遍历数组,当p[i].x-midx的绝对值<d,将p[i]元素放入tmp[]数组                                        
    for (int i = l; i <= r; i++) {
        if (fabs(p[i].x - midx) <= d) {
            tmp[k++] = p[i];
        }
    }
    //对tmp[]临时数组从小到大进行排序 
    sort(tmp, tmp + k, cmp);
    
    //(y) ,i,j指针相当于一个滑动窗口,在tmp[]范围内滑动,滑动窗口大小为1,从左往右滑动
    //找到满足两点y坐标相差距离<=d,求出这两点之间的直线距离,比较这两点之间距离与之前求得的最近点距离哪个小就将它赋值给ans并return 
    for (int i = 0; i < k; i++) {
        for (int j = i + 1; j < k && tmp[j].y - tmp[i].y <= d; j++) {
            double dij = dist(tmp[i], tmp[j]);
            ans = min(ans, dij);
        }
    }
    return ans;// 返回最近点对距离
}

double solve(int l, int r) {
    //递归出口 
    if (l == r) return 1e20;// 如果左右两侧的数组长度为1(即只有一个点),则它们之间的距离应该是无限大
    if (l + 1 == r) return dist(p[l], p[r]);// 如果左右两侧的数组长度为2(即只有两个点),则它们之间的距离可以直接计算

    //分治 
    int mid = (l + r) >> 1;
    double dl = solve(l, mid);//划分左半区 
    double dr = solve(mid + 1, r);//划分右半区 
    double d = min(dl, dr);//找到最近点距离 
    //合并
    return merge(l, r, d);
}

int main() {
    int t, n;
    scanf("%d", &t);
    while (t--) {
        scanf("%d", &n);
        for (int i = 0; i < n; i++) {
            scanf("%lf%lf", &p[i].x, &p[i].y);
        }
        sort(p, p + n, cmp);// 根据x坐标从小到大排序
        printf("%.4lf\n", solve(0, n - 1)); // 求解最近点对距
    }
    return 0;
}

图解:

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值