Cod sursa(job #3236487)

Utilizator memecoinMeme Coin memecoin Data 28 iunie 2024 23:29:39
Problema Adapost Scor 5
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.79 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <cmath>
#include <algorithm>
#include <iomanip>

using namespace std;

const int MAXN = 405;
const double INF = 1e9;
const double EPS = 1e-9;

int N;
double x1[MAXN], y1_[MAXN], x2[MAXN], y2_[MAXN];
double dist[MAXN][MAXN];

double sqr(double x) {
    return x * x;
}

double distance(int i, int j) {
    return sqrt(sqr(x1[i] - x2[j]) + sqr(y1_[i] - y2_[j]));
}

pair<vector<int>, double> hungarian(const vector<vector<double>>& cost) {
    int n = cost.size(), m = cost[0].size();
    vector<double> u(n+1), v(m+1);
    vector<int> p(m+1), way(m+1);
    for (int i = 1; i <= n; ++i) {
        p[0] = i;
        int j0 = 0;
        vector<double> minv(m+1, INF);
        vector<bool> used(m+1, false);
        do {
            used[j0] = true;
            int i0 = p[j0], j1;
            double delta = INF;
            for (int j = 1; j <= m; ++j)
                if (!used[j]) {
                    double cur = cost[i0-1][j-1] - u[i0] - v[j];
                    if (cur < minv[j]) {
                        minv[j] = cur;
                        way[j] = j0;
                    }
                    if (minv[j] < delta) {
                        delta = minv[j];
                        j1 = j;
                    }
                }
            for (int j = 0; j <= m; ++j)
                if (used[j]) {
                    u[p[j]] += delta;
                    v[j] -= delta;
                } else
                    minv[j] -= delta;
            j0 = j1;
        } while (p[j0] != 0);
        do {
            int j1 = way[j0];
            p[j0] = p[j1];
            j0 = j1;
        } while (j0);
    }
    vector<int> assignment(n);
    for (int j = 1; j <= m; ++j)
        assignment[p[j]-1] = j-1;
    return {assignment, -v[0]};
}

pair<double, double> solve() {
    vector<vector<double>> cost(N, vector<double>(N));
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < N; j++) {
            dist[i][j] = distance(i, j);
            cost[i][j] = dist[i][j];
        }
    }

    auto [assignment, _] = hungarian(cost);
    double max_dist = 0;
    for (int i = 0; i < N; i++) {
        max_dist = max(max_dist, dist[i][assignment[i]]);
    }

    for (int i = 0; i < N; i++) {
        for (int j = 0; j < N; j++) {
            cost[i][j] = (dist[i][j] <= max_dist + EPS) ? dist[i][j] : INF;
        }
    }

    auto [final_assignment, sum_dist] = hungarian(cost);

    return {max_dist, sum_dist};
}

int main() {
    ifstream fin("adapost.in");
    ofstream fout("adapost.out");

    fin >> N;
    for (int i = 0; i < N; i++)
        fin >> x1[i] >> y1_[i];
    for (int i = 0; i < N; i++)
        fin >> x2[i] >> y2_[i];

    auto [max_dist, sum_dist] = solve();

    fout << fixed << setprecision(5) << max_dist << " " << sum_dist << endl;

    return 0;
}