模板题,可以用kruskal,也可以用最短路。
#include<cstdio>
#include<string>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn = 5555;
int u[maxn],v[maxn],p[maxn];
double map[105][2];
struct node
{
int id;
double w;
} r[maxn];
int n;
int cmp(const node a,const node b)
{
return a.w < b.w;
}
int find(int x)
{
if(x != p[x])
{
p[x] = find(p[x]);
}
return p[x];
}
double Kruskal(int n,int m)//注意边和结点的数目
{
double ans = 0;
for(int i = 0; i < m; i++) p[i] = i;
sort(r,r+n,cmp);
for(int i = 0; i < n; i++)
{
int e = r[i].id;
int x = find(u[e]);
int y = find(v[e]);
if(x != y)
{
ans += r[i].w;
p[x] = y;
}
}
return ans;
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
for(int i = 0; i < n; i++) scanf("%lf%lf",&map[i][0],&map[i][1]);
int num = 0;
for(int i = 0; i < n; i++)
{
for(int j = i + 1; j < n; j++)
{
u[num] = i;
v[num] = j;
r[num].id = num;
double w1 = map[i][0] - map[j][0];
double w2 = map[i][1] - map[j][1];
r[num++].w = sqrt(w1 * w1 + w2 * w2);
}
}
printf("%.2lf\n",Kruskal(num,n));
if(T) printf("\n");
}
return 0;
}