码迷,mamicode.com
首页 > 其他好文 > 详细

POJ 3625 Building Roads

时间:2014-07-06 09:38:58      阅读:171      评论:0      收藏:0      [点我收藏+]

标签:最小生成树

最小生成树模版题。

N个点,M条已经存在的边。

算出两两点之间的距离,然后Kruskal。


#include<cstdio>
#include<cstring>
#include<string>
#include<queue>
#include<algorithm>
#include<queue>
#include<map>
#include<stack>
#include<iostream>
#include<list>
#include<set>
#include<cmath>
#define INF 0x7fffffff
#define eps 1e-6
using namespace std;
int n,m;
struct lx
{
    int u,v;
    double len;
} l[1001*500];
int fa[1001];
struct node
{
    double x,y;
} point[1001];
int father(int x)
{
    if(x!=fa[x])
        return fa[x]=father(fa[x]);
}
bool cmp(lx a,lx b)
{
    return a.len<b.len;
}
double getlen(node a,node b)
{
    return sqrt(pow(a.x-b.x,2)+pow(a.y-b.y,2));
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(int i=1; i<=n; i++)
        {
            fa[i]=i;
            scanf("%lf%lf",&point[i].x,&point[i].y);
        }
        int cot=0;
        for(int i=1; i<=n; i++)
            for(int j=i+1; j<=n; j++)
            {
                l[cot].u=i;
                l[cot].v=j;
                l[cot++].len=getlen(point[i],point[j]);
            }

        while(m--)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            a=father(a),b=father(b);
            if(a==b)continue;
            fa[b]=a;
        }

        double ans=0;
        sort(l,l+cot,cmp);
        for(int i=0; i<cot; i++)
        {
            int fu=father(l[i].u);
            int fv=father(l[i].v);
            if(fu==fv)continue;
            fa[fv]=fu;
            ans+=l[i].len;

        }

        printf("%.2f\n",ans);
    }
}


POJ 3625 Building Roads,布布扣,bubuko.com

POJ 3625 Building Roads

标签:最小生成树

原文地址:http://blog.csdn.net/dongshimou/article/details/37054761

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!