UVaOJ 10397 - Connect the campus

——by A Code Rabbit

Description

给你一张校园图。

你要铺设线路让校园的每一点都可以连接起来。

校园之前已经有一些旧的线路可供使用了。

要求铺设的线路长度最短。

输入校园图和旧的线路。

输出最短线路的长度。

Types

Graph Algorithms

Analysis

题目可以看出是MST。

Kruskal中的并查集，用来表示两点在不在一个集合中。

事先铺好的线路把两个点连好了，就是该两点已经处于同一集合中了。

我们只要事先根据旧线路将各个点的集合并好，就可以用Kruskal搞定了。

Solution

// UVaOJ 10397
// Connect the Campus
// by A Code Rabbit

#include <algorithm>
#include <cmath>
#include <cstdio>
using namespace std;

const int MAXV = 752;
const int MAXE = MAXV * MAXV;

template <typename T>
struct Point {
    T x, y;
    static double GetDistance(Point a, Point b) {
        return sqrt(pow((a.x - b.x), 2) + pow((a.y - b.y), 2));
    }
};

template <typename T>
struct Edge {
    int u, v;
    T w;
};

template <typename T>
struct Graph {
    Edge<T> edge[MAXE];
    int tot_edge;
    void Init() { tot_edge = 0; }
    void AddEdge(int u, int v, T w) {
        edge[tot_edge].u = u;
        edge[tot_edge].v = v;
        edge[tot_edge].w = w;
        tot_edge++;
    }
};

namespace Kruskal {
    int p[MAXV];
    template <typename T>
    int Cmp(Edge<T> a, Edge<T> b) { return a.w < b.w; }
    int Find(int x) { return p[x] == x ? x : p[x] = Find(p[x]); }
    template <typename T>
    T Go(Edge<T> e[MAXE], int n, int m) {
        //for (int i = 0; i < n; i++) p[i] = i;
        sort(e, e + m, Cmp<T>);
        T ans = 0;
        for (int i = 0; i < m; i++) {
            int u = Find(e[i].u); int v = Find(e[i].v);
            if (u != v) { ans += e[i].w; p[u] = v; }
        }
        return ans;
    }
}

int n, m;
Point<int> point[MAXV];
Graph<double> graph;

int main() {
    while (scanf("%d", &n) != EOF) {
        // Input.
        for (int i = 0; i < n; i++)
            scanf("%d%d", &point[i].x, &point[i].y);
        for (int i = 0; i < n; i++)
            Kruskal::p[i] = i;
        scanf("%d", &m);
        for (int i = 0; i < m; i++) {
            int u, v;
            scanf("%d%d", &u, &v);
            u = Kruskal::Find(u - 1);
            v = Kruskal::Find(v - 1);
            Kruskal::p[u] = v;
        }
        // Compute each the length of edges.
        graph.Init();
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < i; j++) {
                double w = Point<int>::GetDistance(point[i], point[j]);
                graph.AddEdge(i, j, w);
            }
        }
        // Output.
        printf("%.2lf\n", Kruskal::Go(graph.edge, n, graph.tot_edge));
    }
    return 0;
}