poj-2560最小生成树

题目大意:有那么多个点,每个点间都可连线,求出他们的最小生成树。

解题思路:因为每个点间都能够连线,我们又不知道最后需要的是哪条,所以每条边都需要考虑。用sqrt开方求出每两个点间的距离,即他们间的权值,再转化为矩阵图,用prim就能轻松解决啦!

#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstring>
using namespace std;
#define MAX 102
float map[MAX][MAX],MIN,sum,dist[MAX];
int n,flag[MAX];
void prim()//最小生成树的模板
{

    int v;
    int i,j;
    memset(dist,0,sizeof(dist));
    memset(flag,0,sizeof(flag));
    for(j=2; j<=n; j++)
        dist[j]=map[1][j];
    flag[1]=1;
    dist[1]=0;
    for(j=2; j<=n; j++)
    {
        MIN=999999999;
        for(i=1; i<=n; i++)
        {
            if(dist[i]<MIN&&!flag[i]&&dist[i]!=0)
            {
                MIN=dist[i];
                v=i;
            }
        }
        sum+=MIN;
       flag[v]=1;
       for(i=1;i<=n;i++)
       {
           if(!flag[i]&&dist[i]>map[v][i])
              dist[i]=map[v][i];
       }
    }

}
int main()
{
    int i,j;
    float a,b,x[MAX],y[MAX];
    while(scanf("%d",&n)!=EOF)
    {
        sum=0;
        for(i=1; i<=n; i++)
        {
            scanf("%f%f",&a,&b);
            x[i]=a;
            y[i]=b;
            map[i][i]=0;
        }
        for(i=1; i<n; i++)
        {
            for(j=i+1; j<=n; j++)
            {
                map[i][j]=map[j][i]=sqrt((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]));
                //求出两点间的距离,因为每两个点都能连线,所有的点都要算
            }
        }
       prim();
        printf("%.2f\n",sum);
    }
    return 0;
}


你可能感兴趣的:(poj-2560最小生成树)