hdu 4463 Outlets 最小生成树

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
using namespace std;
const double inf=100000000;
double ans,mp[105][105],x[105],y[105],lowcost[105];

int main()
{
    int i,j,k,n,pn,pa;
    while(~scanf("%d",&n))
    {
        if(n==0) break;
        scanf("%d%d",&pn,&pa);
        for(i=1;i<=n;i++)
            scanf("%lf%lf",&x[i],&y[i]);

        for(i=1;i<=n;i++)
            for(j=1;j<=n;j++)
                mp[i][j]=sqrt((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]));
        ans=mp[pn][pa];
        mp[pn][pa]=0;
        mp[pa][pn]=0;
        for(i=1;i<=n;i++)
            lowcost[i]=mp[1][i];
        lowcost[1]=-1;

        for(i=1;i<=n;i++)
        {
            int v=-1;
            double minx=inf;
            for(j=1;j<=n;j++)
            {
                if(lowcost[j]!=-1&&minx>lowcost[j])
                {
                    minx=lowcost[j];
                    v=j;
                }
            }
            if(v!=-1)
            {
                ans+=minx;
                lowcost[v]=-1;
                for(j=1;j<=n;j++)
                {
                    if(mp[v][j]<lowcost[j])
                        lowcost[j]=mp[v][j];
                }
            }
        }
        printf("%.2lf\n",ans);
    }
    return 0;
}

版权声明：本文为博主原创文章，未经博主允许不得转载。