HDOJ  1162   Eddy's picture

题目：http://acm.hdu.edu.cn/showproblem.php?pid=1162

prime算法好些，但Kruskal也可以

#include <stdio.h>
#include <math.h>
#include <string.h>
#define MAX 1000000
double Edge[110][110];
double as[110][2];
double lowcost[110];
int n;
void init()
{
    int i,k;
    for(i=0;i<n;i++)
        scanf("%lf%lf",&as[i][0],&as[i][1]);
    for(i=0;i<n;i++)
    {
        for(k=0;k<n;k++)
        {
            if(i==k)
                Edge[i][k]=MAX;
            else
                Edge[i][k]=sqrt((as[i][0]-as[k][0])*(as[i][0]-as[k][0])+(as[i][1]-as[k][1])*(as[i][1]-as[k][1]));
        }
    }
    memset(lowcost,0,sizeof(lowcost));
}
void prim()
{
    int i,k;
    double sum=0;
    lowcost[0]=-1;
    for(i=1;i<n;i++)
        lowcost[i]=Edge[0][i];
    for(i=1;i<n;i++)
    {
        double min=MAX;
        int j;
        for(k=0;k<n;k++)
        {
            if(lowcost[k]!=-1&&lowcost[k]<min)
                j=k,min=lowcost[k];
        }
        sum+=min;
        lowcost[j]=-1;
        for(k=0;k<n;k++)
            if(Edge[j][k]<lowcost[k])
                lowcost[k]=Edge[j][k];
    }
    printf("%0.2lf\n",sum);
}
int main()
{
    while(scanf("%d",&n)!=EOF)
        init(),prim();
    return 0;
}
**************************************************
#include <stdio.h>
#include <math.h>
#include <stdlib.h>
#include <string.h>
#define MAX 1000000
#define MAM 105
struct edge
{
    int u,v;
    double w;
}edges[MAX];
double x[MAM],y[MAM];
int parent[MAX];
int n,m;
int i,j;
double sumweight;
int Find(int x)
{
    int s;
    for(s=x;parent[s]>=0;s=parent[s]);
    while(s!=x)
    {
        int tmp=parent[x];
        parent[x]=s;
        x=tmp;
    }
    return s;
}
void Union(int R1,int R2)
{
    int r1=Find(R1),r2=Find(R2);
    int tmp=parent[r1]+parent[r2];
    if(parent[r1]>parent[r2])
    {
        parent[r1]=r2;
        parent[r2]=tmp;
    }
    else
    {
        parent[r2]=r1;
        parent[r1]=tmp;
    }
}
int cmp(const void *a,const void *b)
{
    edge aa=*(const edge *)a;
    edge bb=*(const edge *)b;
    if(aa.w>bb.w)
        return 1;
    else
        return -1;
}
void Kruskal()
{
    int num=0;
    int u,v;
    memset(parent,-1,sizeof(parent));
    for(i=0;i<m;i++)
    {
        u=edges[i].u;
        v=edges[i].v;
        if(Find(u)!=Find(v))
        {
            sumweight+=edges[i].w;
            num++;
            Union(u,v);
        }
        if(n-1<=num)
            break;
    }
}
int main()
{
    double d;
    while(scanf("%d",&n)!=EOF)
    {
        for(i=0;i<n;i++)
            scanf("%lf%lf",&x[i],&y[i]);
        int mi=0;
        for(i=0;i<n;i++)
        {
            for(j=i+1;j<n;j++)
            {
                d=sqrt((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]));
                edges[mi].u=i;
                edges[mi].v=j;
                edges[mi].w=d;
                mi++;
            }
        }
        m=mi;
        qsort(edges,m,sizeof(edges[0]),cmp);
        sumweight=0.0;
        Kruskal();
        printf("%0.2lf\n",sumweight);
    }
    return 0;
}