POJ1287 Networking【最小生成树】

题意：

给出n个节点，再有m条边，这m条边代表从a节点到b节点电缆的长度，现在要你将所有节点都连起来，并且使长度最小

思路：

这是个标准的最小生成树的问题，用prim的时候需要注意的是他有重边，取边最小的那条加入图里就可以了，但是kruskal可以忽略这个问题

代码：

prim：

#include <iostream>
#define maxn 55
#define inf 1<<29
using namespace std;

int map[maxn][maxn];
int n,m;

void prim()
{
    int d[maxn],vis[maxn];
    int i,v,j,minn;
    for(i=1;i<=n;i++){
        vis[i]=0;
        d[i]=map[1][i];
    }
    for(i=1;i<=n;i++)
    {
        minn=inf;
        for(j=1;j<=n;j++)
            if(!vis[j] && d[j]<minn)
            {
                minn=d[j];
                v=j;
            }
        vis[v]=j;
        for(j=1;j<=n;j++)
        {
            if(!vis[j] && map[v][j]<d[j])
                d[j]=map[v][j];
        }
    }
    for(d[0]=0,i=1;i<=n;i++)
       d[0]+=d[i];
    cout<<d[0]<<endl;
}

int main()
{
    int i,j,a,b,c;
    while(cin>>n,n)
    {
        cin>>m;
        for(i=0;i<=n;i++)
        {
            for(j=0;j<=n;j++)
                if(i!=j) map[i][j]=inf;
                else map[i][j]=0;
        }
        for(i=0;i<m;i++)
        {
            cin>>a>>b>>c;
            if(map[a][b]>c) map[a][b]=map[b][a]=c;
        }
        prim();
    }
    return 0;
}

View Code