题目大意:
大学要铺设电线,在n个建筑中,使得这n个建筑,直接或间接相连,并给你了这n个建筑的坐标,还有m条已经相连的边,这些边用起点和终点表示。现在要你求怎么铺设电线才能使得总距离最小。
大学要铺设电线,在n个建筑中,使得这n个建筑,直接或间接相连,并给你了这n个建筑的坐标,还有m条已经相连的边,这些边用起点和终点表示。现在要你求怎么铺设电线才能使得总距离最小。
解析:最小生成树题,因为有些边已经铺设了电线,所以我们将这些边权值设为了0。然后就套kruskal算法模板。
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
const int INF = 0x3f3f3f3f;
const int N = 760;
struct Edge {
int e,s;
double v;
}a[N*N/2];
struct Point {
double x,y;
}p[N];
int n , m, pa[N];
int vis[N][N];
double dis(Point a,Point b) {
return sqrt((a.x - b.x) * (a.x - b.x) + (a.y - b.y) * (a.y - b.y));
}
bool cmp(Edge a,Edge b) {
return a.v < b.v;
}
int find(int x) {
if(x == pa[x]) {
return x;
}else {
return find(pa[x]);
}
}
void init() {
for(int i = 0; i <= n; i++) {
pa[i] = i;
}
}
int main() {
while(scanf("%d",&n) != EOF) {
for(int i = 1; i <= n; i++) {
scanf("%lf%lf",&p[i].x,&p[i].y);
}
int u,v;
memset(vis,0,sizeof(vis));
scanf("%d",&m);
for(int i = 0; i < m; i++) {
scanf("%d%d",&u,&v);
vis[u][v] = vis[v][u] = true;
}
int tot = 0;
for(int i = 1; i <= n; i++) {
for(int j = i + 1; j <= n; j++) {
if(vis[i][j]) {
a[tot].s = i;
a[tot].e = j;
a[tot].v = 0;
}else {
a[tot].s = i;
a[tot].e = j;
a[tot].v = dis(p[i],p[j]);
}
tot++;
}
}
init();
sort(a,a+tot,cmp);
int num = 0;
double sum = 0;
for(int i = 0; i < tot && num < n-1; i++) {
int k = find(a[i].s) , g = find(a[i].e);
if(k != g) {
pa[g] = k;
sum += a[i].v;
num++;
}
}
printf("%.2lf\n",sum);
}
return 0;
}