POJ 3625 Building Roads

最小生成树模版题。

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);
    }
}