P2872 [USACO07DEC]Building Roads S
-
看题目描述:我们可以看出这是一道关于最小生成树的问题,然后有两个算法可以选择,prim和kruskal
因为不会别的 -
再看一下m是题目里自己连好的边的数量,让求把其它点连进去的最小的权重和,此时就可以分析两种算法的优缺点了。prim是加点,通过bool数组的值确定此点是否已经在集合里了,然后再进行下一步,因此,我们可以想到,如果用这种方法就可能产生多个连通分支,并不能构成一棵树,所以使用prim算法依然需要使用并查集去维护是否可以构成一棵树,显然不是最优的。那就只能选择kruskal算法,通过加边来构成最小生成树。
-
然后写好了,提交可能会报re是因为数组开小了不能只开1000,因为它有1000个点最多,所以它边数最多可以达到(1e6 - 1000)/2条,那么我们就把数组开到1e6.因为精度问题第2,8,9,10的样例会wa掉,所以需要在算距离的函数里加入3个强转来保障精度。
-
最后你就可以ac这道题目了。
ac代码在下面
#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
const int N = 1e6 + 10,M = 1010;
using namespace std;
int n,m;
int p[M];
struct Edge
{
int a ,b ;
double w;
bool operator< (const Edge &W)const
{
return w < W.w;
}
}edges[N];
struct Point
{
int x,y;
}po[M];
double dist(Point a, Point b)
{
return (double)sqrt((double)(a.x - b.x) * (a.x - b.x) + (double)(a.y - b.y) * (a.y - b.y));
}
int find(int x)
{
if(p[x] != x) p[x] = find(p[x]);
return p[x];
}
int main()
{
scanf("%d%d",&n,&m);
for(int i = 1;i <= n;i ++) p[i] = i;
for(int i = 1;i <= n;i ++) scanf("%d%d",&po[i].x,&po[i].y);
int cnt = 0;
for(int i = 1;i <= n;i ++)
for(int j = i + 1;j <= n;j ++)
edges[cnt ++] = {i,j,dist(po[i],po[j])};
while(m --)
{
int a ,b;
scanf("%d%d",&a,&b);
a = find(a),b = find(b);
if(a != b) p[a] = b;
}
sort(edges,edges + cnt);
double res = 0;
for(int i = 0;i < cnt;i++)
{
int a = edges[i].a , b = edges[i].b ;
double w = edges[i].w ;
a = find(a),b = find(b);
if(a != b)
{
p[a] = b;
res += w;
}
}
printf("%.2f\n",res);
return 0;
}