图结构练习——最小生成树（Prim算法）

题目描述

 有n个城市，其中有些城市之间可以修建公路，修建不同的公路费用是不同的。现在我们想知道，最少花多少钱修公路可以将所有的城市连在一起，使在任意一城市出发，可以到达其他任意的城市。

 

输入

 输入包含多组数据，格式如下。

第一行包括两个整数n m，代表城市个数和可以修建的公路个数。(n<=100)

剩下m行每行3个正整数a b c，代表城市a 和城市b之间可以修建一条公路，代价为c。

 

输出

 每组输出占一行，仅输出最小花费。

示例输入

3 2
1 2 1
1 3 1
1 0

示例输出

2
0

代码：

#include <stdio.h>
#include <string.h>
#define inf  999999
int tu[101][101];
int biao[101];
int dis[101];
void prim(int n);
int sum=0;
int main()
{
    int n,m,i,j,a,b,c;
    while(scanf("%d %d",&n,&m)!=EOF)


    {
        sum=0;


        for(i=1; i<=n; i++)
        {
            for(j=1; j<=n; j++)
                tu[i][j]=inf;
            // tu[i][i]=0;
        }


        for(i=1; i<=m; i++)
        {
            scanf("%d %d %d",&a,&b,&c);
            if(tu[a][b]>c)  //这一句不可少，此题特殊，如输入1 2 1后还可以输入1 2 2,所以以后写
           tu[a][b]=tu[b][a]=c;//同类程序时要注意加上这一句。有的题只输入一组，是单联通，此题涉及 
        }                     //多联通，必须找最小的入图。所以不论题目怎样，都加上这一句，准不错。
        prim(n);
        printf("%d\n",sum);
    }
    return 0;
}


void  prim (int n)
{
    memset(biao,0,sizeof(biao));
    int i,j;
    int pos;
    for(i=1; i<=n; i++)
        dis[i]=tu[1][i];
    biao[1]=1;
    for(i=1; i<=n-1; i++)
    {
        int min=inf;
        for(j=1; j<=n; j++)
            if(dis[j]<min&&!biao[j])
            {
                min=dis[j];
                pos=j;
            }
        if(min>=inf)
        {
            break;
        }
        sum=sum+min;
        biao[pos]=1;


        for(j=1; j<=n; j++)
            if(!biao[j]&&tu[pos][j]<dis[j])
                dis[j]=tu[pos][j];




    }


    return ;
}