UVa 10397 - Connect the Campus

因为已经有部分建筑用电线连起来了，要求还需要多少电线才可以将全部建筑连起来，只需要将用电线连起来的城市之间的距离设为0，再用Kruskal。

代码如下：

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <cmath>
using namespace std;
const int DMAXN = 755;
const int EMAXN = 281630;
int n, p[DMAXN], u[EMAXN], v[EMAXN], r[EMAXN];
double x[DMAXN], y[DMAXN], dis[EMAXN], ans;
bool vis[DMAXN][DMAXN];
int cmp(const int i, const int j)
{
return dis[i] < dis[j];
}
int find(int i)
{
return p[i] == i ? i : (p[i] = find(p[i]));
}
void Kruskal()
{
ans = 0;
sort(r, r+n, cmp);
for(int i=0; i<n; i++)
{
int e = r[i];
int xx = find(u[e]);
int yy = find(v[e]);
if(xx != yy)
{
ans += dis[e];
p[xx] = yy;
}
}
}
int main()
{
#ifdef test
freopen("sample.txt", "r", stdin);
#endif
int dot, cable, fx, fy;
while(scanf("%d", &dot) != EOF)
{
n = 0;
double d1, d2;
memset(vis, false, sizeof(vis));
for(int i=0; i<dot; i++)
{
scanf("%lf%lf", &x[i], &y[i]);
p[i] = i;
}
scanf("%d", &cable);
for(int i=0; i<cable; i++)
{
scanf("%d%d", &fx, &fy);
--fx;
--fy;
vis[fx][fy] = vis[fy][fx] = true;
}
for(int i=0; i<dot; i++)
for(int j=i+1; j<dot; j++)
{
u
= i;
v
= j;
r
= n;
if(vis[i][j])
dis[n++] = 0.0;
else
{
d1 = x[i] - x[j];
d2 = y[i] - y[j];
dis[n++] = sqrt(d1*d1+d2*d2);
}
}
Kruskal();
printf("%.2lf\n", ans);
}
return 0;
}