hdu Eddy's picture

直接用模板，prim kruskual 都可以。

 

prim:

#include<stdio.h>
#include<math.h>
#include<string.h>
double lowcost[110];
double dp[110][110];
int visit[110];
int n;

double prim()
{
    int j,z;
    double min,sum=0.0;
    int yes;
    memset(visit,0,sizeof(visit));
    visit[1]=1;
    for(;;)
    {
        min=1000000.0;
        yes=-1;
        for(j=1;j<=n;j++)
            if(lowcost[j]<min && !visit[j])
            {
                min=lowcost[j];
                z=j;
                yes=1;
            }
        if(yes==-1) break;
        visit[z]=j;
        sum+=min;
        for(j=1;j<=n;j++)
            if(lowcost[j]>dp[z][j] && !visit[j]) lowcost[j]=dp[z][j];
    }
    return sum;
}
int main()
{
    int i,j;
    double x[110],y[110];
    while(scanf("%d",&n)!=EOF)
    {
        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++)
                dp[i][j]=sqrt(pow((x[i]-x[j]),2)+pow((y[i]-y[j]),2));
        for(i=1;i<=n;i++)
            lowcost[i]=dp[1][i];
        printf("%.2lf\n",prim());
    }
    return 0;
}

kruskual;

#include<stdio.h>
#include<math.h>
#include<algorithm>
using namespace std;
double sum;
int coun;

struct node
{
    int strat,end;
    double len;
}list[11000];
bool cmp(node a,node b)
{
    return a.len < b.len ;
}

int father[11000];

int find(int a)
{
    if(a==father[a]) return a;
    else return find(father[a]);
}

void bing(int a,int b)
{
    father[find(a)]= father[find(b)];
}

int main()
{
    int n,i,j;
    double x[110],y[110];
    while(scanf("%d",&n)!=EOF)
    {
        for(i=1;i<=n;i++)
            scanf("%lf%lf",&x[i],&y[i]);
        coun=0;
        for(i=1;i<n;i++)
            for(j=i+1;j<=n;j++)
            {
                coun++;
                list[coun].strat = i;
                list[coun].end = j;
                list[coun].len = sqrt(pow((x[i]-x[j]),2)+pow((y[i]-y[j]),2));
            }
        for(i=1;i<=coun;i++) father[i]=i;
        sort(list+1,list+coun+1,cmp);
        sum=0;
        for(i=1;i<=coun;i++)
            if(find(list[i].strat ) != find(list[i].end ))
            {
                bing(list[i].strat ,list[i].end );
                sum+=list[i].len ;
            }
        printf("%.2lf\n",sum);
    }
    return 0;
}