洛谷P7883 平面最近点对

最新推荐文章于 2023-08-27 15:48:36 发布

stdforces

最新推荐文章于 2023-08-27 15:48:36 发布

阅读量228

点赞数

文章标签：平面算法贪心算法

本文链接：https://blog.csdn.net/stdforces/article/details/126354365

版权

题意：

给出平面上的 $n$ 个整点，求出最近两个点的距离的平方。

方法：

分治处理，每次递归左半边区间，找出这个区间的点的距离最小值的平方，随后考虑怎么合并。

贪心考虑点，假设当前得到的最小答案为 $an s$ ，那么显然水平距离和竖直距离 $\geq ans$ 的我们一定不考虑，所以剔除这些点，然后暴力更新答案。

如果分治+快速排序的话时间复杂度是 $O(nlog^{2}n)$ 的，如果是归并排序合并时处理能达到 $O (n l o g n)$ ，但是难写，利用 $se t$ 可以两全其美

我们只考虑向前 $an s$ 的，因为 $a$ 到 $b$ 距离和 $b$ 到 $a$ 距离是等价的，先按 $x$ 排序，能保证找到所有前面的点，加入一个点时，我们就从没删去的第一个点开始删，直到水平距离 $< an s$ 。在纵向距离，用迭代器找到上面的点和下面的点，暴力判断。

那么暴力判断又如何保证时间复杂度是 $O (n l o g n)$ 的呢，接下来证明暴力判断的时间复杂度是 $O (n)$ 的。

不过这个做法由于 $y$ 间距超过 $an s$ 不被删去，所以会被 $x = k$ 的直线卡，比较好的解决方法是将所有点转过一个不那么熟悉的角度即可。

#include<bits/stdc++.h>
#define ll long long
using namespace std;

struct vec
{
    ll x,y;
    vec operator-(const vec& a) const {return (vec){x-a.x,y-a.y};}
    bool operator<(const vec& a) const
    {
        if(y==a.y) return x<a.x;
        return y<a.y;
    }
    friend istream& operator>>(istream& o,const vec& a)
    {
        scanf("%lld%lld",&a.x,&a.y);
        return o;
    }
};

ll getlen(const vec& a)
{
    return a.x*a.x+a.y*a.y;
}

int n;
set<vec>set1;
vector<vec>v(500005);

int main()
{
    cin>>n;
    for(int i=1;i<=n;i++) cin>>v[i];
    sort(v.begin()+1,v.begin()+1+n,[&](const vec& a,const vec& b){
        if(a.x==b.x) return a.y<b.y;
        return a.x<b.x;
    });
    ll ans=getlen(v[1]-v[2]);
    for(int i=1,j=0;i<=n;i++)
    {
        set1.insert(v[i]);
        while((v[j+1].x-v[i].x)*(v[j+1].x-v[i].x)>=ans) set1.erase(v[++j]);
        auto it=set1.lower_bound(v[i]); it++;
        while(it!=set1.end()&&(v[i].y-it->y)*(v[i].y-it->y)<ans)
        {
            ans=min(ans,getlen(*it-v[i]));
            it++;
        }
        it=set1.lower_bound(v[i]);
        if(it!=set1.begin())
        {
            it--;
            while((v[i].y-it->y)*(v[i].y-it->y)<ans)
            {
                ans=min(ans,getlen(*it-v[i]));
                if(it==set1.begin()) break;
                it--;
            }
        }
    }
    cout<<ans;
      return 0;
}