原题:
http://172.16.0.132/senior/#contest/show/1904/0
题目描述:
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个用空格隔开的整数:X_i、Y_i
- 第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选择在农场1和农场2间建一条长度为2.00的道路,在农场3和农场4间建一条长度为2.00的道路。这样,所建道路的总长为4.00,并且这是所有方案中道路总长最小的一种。
分析:
整个地图看做一张二维表,然后给出的是每一个农场即顶点的坐标。我们不难想出:对于任意两个顶点,我们都可以直接修一条路,这条路的权值就是两个顶点间的欧几里得距离(d=sqrt((x1-x2)(x1-x2)+(y1-y2)(y1-y2)))。按照Prim算法或者Kruskal算法求出最小生成树的权值和即可。需要注意的是,有的农场之间已经存在了一条或多条路径,对算法形成了一定的干扰。
我们不需要考虑已经存在原有路径的a,b两个端点又向边集数组加入了一条连接这两个点的路径,因为进行sort排序时,我们按照的是权值由小到大排序,权值为0的会优先加入到空边图当中。
实现:
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
int n,m,fa[1002],zz;
struct dian{int x,y;} d[1002];
struct bian{int frm,to;double v;} e[1000002];
double ans;
bool kp(const bian &i,const bian &j) {return i.v<j.v;}
int find(int x)
{
if(fa[x]!=x) fa[x]=find(fa[x]);
return fa[x];
}
double fen(int i,int j)
{
long long X=(long long)d[i].x-d[j].x,Y=(long long)d[i].y-d[j].y;
return sqrt(X*X+Y*Y);
}
void cha(int x,int y)
{
double z=fen(x,y);
zz++;
e[zz].frm=x;
e[zz].to=y;
e[zz].v=z;
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d%d",&d[i].x,&d[i].y);
fa[i]=i;
}
for(int i=1;i<=m;i++)
{
int x,y,r1,r2;
scanf("%d%d",&x,&y);
r1=find(x);
r2=find(y);
fa[r2]=r1;
}
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
if(i==j) continue;
int r1=find(i),r2=find(j);
if(r1!=r2) cha(i,j);
}
sort(e+1,e+zz+1,kp);
for(int i=1;i<=zz;i++)
{
int r1,r2;
r1=find(e[i].frm); r2=find(e[i].to);
if(r1!=r2)
{
ans+=e[i].v;
fa[r2]=r1;
}
}
printf("%.2lf",ans);
}