UVA 10034 Freckles （最小生成树模板）

题意：给出n个点的坐标，求将所有点连接在一起的最小代价。

思路：prim算法

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <cmath>
#define INF 0x7fffffff

using namespace std;
double g[105][105],dis[105],a[105],b[105];
int vis[105],T,n;

double prim(int x)
{
    double ans=0;
    vis[x]=1;
    for(int i=0;i<n;i++)
    {
        if(vis[i]==0)
        {
            dis[i]=g[x][i];
        }
    }
    for(int i=1;i<n;i++)
    {
        double mi=INF;
        int k;
        for(int j=0;j<n;j++)
        {
            if(vis[j]==0&&mi>dis[j])
            {
                mi=dis[j];
                k=j;
            }
        }
        ans+=mi;
        //printf("%f\n",ans);
        vis[k]=1;
        for(int j=0;j<n;j++)
        {
            if(vis[j]==0&&dis[j]>g[k][j])
            {
                dis[j]=g[k][j];
            }
        }
    }
    return ans;
}
int main()
{
    scanf("%d",&T);
    for(int j=1;j<=T;j++)
    {
        memset(vis,0,sizeof(vis));
        scanf("%d",&n);
        for(int i=0;i<n;i++)
            scanf("%lf%lf",&a[i],&b[i]);
        for(int i=0;i<n-1;i++)
        {
            for(int j=i+1;j<n;j++)
            {
                g[i][j]=g[j][i]=sqrt(pow((a[i]-a[j]),2)+pow((b[i]-b[j]),2));
            }
        }
        printf("%.2f\n",prim(0));
        if(j!=T)
            printf("\n");
    }
    return 0;
}