Cod sursa(job #3145285)

Utilizator Mihai_EduardMihai Eduard Mihai_Eduard Data 14 august 2023 17:23:11
Problema Cele mai apropiate puncte din plan Scor 10
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.7 kb
#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef long double ld;

#define pb push_back

struct Point{ll x; ll y;};
bool comp_x(Point &A, Point &B){return A.x<B.x;}
bool comp_y(Point &A, Point &B){return A.y<B.y;}

ll get_distance(Point A, Point B){
    return (A.x-B.x)*(A.x-B.x)+(A.y-B.y)*(A.y-B.y);
}

ll bruteforce(vector<Point> points, int st, int dr){
    ll minim = LLONG_MAX;
    for(int i=st;i<=dr;i++) for(int j=i+1;j<=dr;j++){
            minim=min(minim, get_distance(points[i],points[j]));
    }
    return minim;
}

ll find_closest(vector<Point> x_sorted, int st, int dr){
    if(dr-st+1<=3)
        return bruteforce(x_sorted, st, dr);

    int mid=(st+dr)/2;
    ll dist=min(find_closest(x_sorted,st,mid),find_closest(x_sorted,mid+1,dr));

    vector<Point> y_sorted;
    for(int i=st;i<=dr;i++){
        if((x_sorted[i].x-x_sorted[mid].x)*(x_sorted[i].x-x_sorted[mid].x)<=dist)
            y_sorted.push_back(x_sorted[i]);
    }

    sort(y_sorted.begin(), y_sorted.end(), comp_y);
    for(int i=0;i<(int)y_sorted.size();i++){
        for(int j=i+1;j<(int)y_sorted.size() && (y_sorted[j].y-y_sorted[i].y)*(y_sorted[j].y-y_sorted[i].y)<dist;j++){
            dist=min(dist, get_distance(y_sorted[i],y_sorted[j]));
        }
    }

    return dist;
}

void solve(){
    freopen("cmap.in", "r", stdin);
    freopen("cmap.out", "w", stdout);

    int N; cin>>N;
    vector<Point> x_sorted;
    for(int i=0;i<N;i++){
        ll x, y; cin>>x>>y;
        x_sorted.push_back({x,y});
    }
    sort(x_sorted.begin(), x_sorted.end(), comp_x);

    cout<<fixed<<setprecision(10)<<sqrt(find_closest(x_sorted, 0, N-1));
}

int main()
{
    ios_base::sync_with_stdio(false); cin.tie(0);
    solve();
    return 0;
}