分治法求解平面上最接近的点对

平面上的最接近点对

题目描述

给定平面上 $n$ 个点，找出其中的一对点的距离，使得在这 $n$ 个点的所有点对中，该距离为所有点对中最小的。

输入格式

第一行一个整数 $n$，表示点的个数。

接下来 $n$ 行，每行两个实数 $x,y$ ，表示一个点的行坐标和列坐标。

输出格式

仅一行，一个实数，表示最短距离，四舍五入保留 $4$ 位小数。

样例 #1

样例输入 #1

3
1 1
1 2
2 2

样例输出 #1

1.0000

提示

数据规模与约定

对于 $100\%$ 的数据，保证 $1\le n\le 10^4$，$0\le x,y\le 10^9$，小数点后的数字个数不超过 $6$。

题解：

#include <iostream>
#include <cmath>
#include <iomanip>
#include <algorithm>
using namespace std;
int temp[10001];
struct point {
    double x, y;
}p[10001];

double dis(int i, int j) {
    return sqrt((p[i].x - p[j].x) * (p[i].x - p[j].x) + (p[i].y - p[j].y) * (p[i].y - p[j].y));
}

bool cmp(point a, point b) {
    if (a.x == b.x)
        return a.y < b.y;
    else
        return a.x < b.x;
}

bool cmp_y(int i, int j) {
    return p[i].y < p[j].y;
}

double merge(int l, int r) {
    if (l == r)
        return 9999999997;
    if (l + 1 == r)
        return dis(l, r);
    int mid = (l + r) >> 1;
    double d1 = merge(l, mid);
    double d2 = merge(mid + 1, r);
    double dist = d1 < d2 ? d1 : d2;
    int k = 0;
    for (int i = l; i <= r; i++) {
        if (fabs(p[i].x - p[mid].x) <= dist)
            temp[k++] = i;
    }
    //按y的大小排序
    sort(temp, temp + k, cmp_y);
    for (int i = 0; i < k; i++) {
        for (int j = i + 1; j < k && p[temp[j]].y - p[temp[i]].y < dist; j++) {
            dist = min(dist, dis(temp[i], temp[j]));
        }
    }
    return dist;
}

int main()
{
    int n; cin >> n;
    for (int i = 0; i < n; i++) {
        cin >> p[i].x >> p[i].y;
    }
    if (n == 1) {
        cout << fixed << setprecision(4) << 0.0f;
        return 0;
    }
    //按照x坐标的大小排序
    sort(p, p + n, cmp);
    cout << fixed << setprecision(4) << merge(0, n - 1);
    return 0;
}

代码思路：

1. 将所有点按照x坐标大小排序。
2. 按照二分的思想，把数组二分为两半，则这两个点要么在同一侧，要么分布在两侧。
3. 二分的终点是l与r相邻但不相等。
4. 求出左侧最小点对和右侧最小点对中的最小值，那么针对一个点在左侧，一个在右侧 中的情况，将中间点的 x 坐标作为 mid。划分一个 [mid - dis, mid + dis]区域，如果存在最小距离点对，必定存在这个区域中，将符合要求的点的编号存入temp数组。
5. 再将temp数组按照y坐标大小排序。
6. 然后用双指针暴力枚举出最小点对的距离即可，将最小距离返回。