——by A Code Rabbit
给你一张校园图。
你要铺设线路让校园的每一点都可以连接起来。
校园之前已经有一些旧的线路可供使用了。
要求铺设的线路长度最短。
输入校园图和旧的线路。
输出最短线路的长度。
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;
}