问题描述
Farmer John最近得到了一些新的农场,他想新修一些道路使得他的所有农场可以经过原有的或是新修的道路互达(也就是说,从任一个农场都可以经过一些首尾相连道路到达剩下的所有农场)。有些农场之间原本就有道路相连。
所有N(1 <= N <= 1,000)个农场(用1..N顺次编号)在地图上都表示为坐标为(X_i, Y_i)的点(0 <= X_i <= 1,000,000;0 <= Y_i <= 1,000,000),两个农场间道路的长度自然就是代表它们的点之间的距离。现在Farmer John也告诉了你农场间原有的M(1 <= M <= 1,000)条路分别连接了哪两个农场,他希望你计算一下,为了使得所有农场连通,他所需建造道路的最小总长是多少。
输入格式
- 第1行: 2个用空格隔开的整数:N 和 M
* 第2..N+1行: 第i+1行为2个用空格隔开的整数:Xi、Yi
* 第N+2..N+M+2行: 每行用2个以空格隔开的整数i、j描述了一条已有的道路,这条道路连接了农场i和农场j
输出格式
- 第1行: 输出使所有农场连通所需建设道路的最小总长,保留2位小数,不必做任何额外的取整操作。为了避免精度误差,计算农场间距离及答案时请使用64位实型变量
样例输入
4 1
1 1
3 1
2 3
4 3
1 4
样例输出
4.00
提示
输入说明:
FJ一共有4个坐标分别为(1,1),(3,1),(2,3),(4,3)的农场。农场1和农场
4之间原本就有道路相连。
输出说明:
FJ选择在农场1和农场2间建一条长度为2.00的道路,在农场3和农场4间建一
条长度为2.00的道路。这样,所建道路的总长为4.00,并且这是所有方案中道路
总长最小的一种。
非常明显的一道最小生成树进阶题,非常适合刚刚打完模板的新手们进行提升。
本题与模板最大的一个区别就是从编号变为二维,还有帮你已经建好了几条道路。
每一行输入的都是二维坐标,因此可以开双重循环,记录每一条路之间的距离。至于两点之间的距离自然是勾股定理的灵活运用:a*a+b*b=c*c。
题目中提到了会帮你提前修好一些道路,你就只用将它们单独处理一遍即可。
最后,就是并查集的合并了,步骤和模板并无大差。
只不过要注意,最后的输出要保留二位小数。可以像我一样cout<<fixed<<setprecision(2)<<ans,也可以printf("%llf",ans)。
#include<bits/stdc++.h>
using namespace std;
int n,m,cnt,f[5000010],sum;
double ans;
struct note
{
int x,y;
} a[5000010];
struct node
{
int x,y;
double w;
} e[5000010];
bool cmp(node x,node y)
{
if(x.w==y.w) return x.x<y.x;
return x.w<y.w;
}
int find(int x)
{
return x==f[x]?x:f[x]=find(f[x]);
}
int main()
{
scanf("%d%d",&n,&m);
for ( int i=1; i<=n; i++)
{
scanf("%d%d",&a[i].x,&a[i].y);
f[i]=i;
}
for(int i=1;i<=n;i++)
{
for(int j=i+1;j<=n;j++)
{
e[++cnt].x=i;
e[cnt].y=j;
e[cnt].w=(double)sqrt((double)(a[i].x-a[j].x)*(double)(a[i].x-a[j].x)+(double)(a[i].y-a[j].y)*(double)(a[i].y-a[j].y));//距离
}
}
for(int i=1;i<=m;i++)
{
int x,y;
scanf("%d%d",&x,&y);
e[++cnt].x=x;
e[cnt].y=y;
e[cnt].w=0.0;
}//单独处理
sort(e+1,e+1+cnt,cmp);
for(int i=1;i<=cnt&&sum<n-1; i++)
{
int x=find(e[i].x);
int y=find(e[i].y);
if(x!=y)
{
f[x]=y;
sum++;
ans+=e[i].w;
}
}
cout<<fixed<<setprecision(2)<<double(ans);
return 0;
}