题意
给出每个点的坐标,求这个图中的最小生成树。
思路
比起普通的最小生成树,我们这题要用勾股定律求出两个点之间的直线距离,剩下的最小生成树我们用并查集来做就好了。
代码
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
int m,n;
double x[101],y[101],father[9604],ans;
struct node{
int l,r;
double v;
}s[9604];
double value(int x1,int y1) {return sqrt(pow(y[x1]-y[y1],2)+pow(x[x1]-x[y1],2));}//这里求两点之间的直线距离
int cmp(node x,node y) {return x.v<y.v;}
int find(int x)
{
int f1=x,f2;
while (x!=father[x]) x=father[x];
while (f1!=father[f1])
{
f2=father[f1];
father[f1]=x;
f1=f2;
}
return x;
}
int main()
{
scanf("%d",&n);
for (int i=1;i<=n;i++)
scanf("%lf%lf",&x[i],&y[i]),father[i]=i;
for (int i=1;i<n;i++)
for (int j=i+1;j<=n;j++)
{
s[++m].v=value(i,j);
s[m].l=i;
s[m].r=j;
}
sort(s+1,s+m+1,cmp);
for (int i=1;i<=m;i++)//用并查集求最小生成树
{
int a=find(s[i].l),b=find(s[i].r);
if (a!=b)
{
father[b]=a;
ans+=s[i].v;
}
}
printf("%.2lf",ans);
}