数据结构实验之图论九：最小生成树

Problem Description
有n个城市，其中有些城市之间可以修建公路，修建不同的公路费用是不同的。现在我们想知道，最少花多少钱修公路可以将所有的城市连在一起，使在任意一城市出发，可以到达其他任意的城市。
Input
输入包含多组数据，格式如下。
第一行包括两个整数n m，代表城市个数和可以修建的公路个数。(n <= 100， m <=10000)
剩下m行每行3个正整数a b c，代表城市a 和城市b之间可以修建一条公路，代价为c。
Output
每组输出占一行，仅输出最小花费。
Sample Input
3 2
1 2 1
1 3 1
1 0
Sample Output
2
0
Hint
Source
赵利强

#include<string.h>
#include<stdio.h>
#include<math.h>
#define inf 0x3f3f3f
int s[110][110];
int dis[150];//dis[i]记录最小边权值，i记录顶点
int sum,flag;
int f[110];
void prim(int n)
{
    int i,j,k;
    for(i = 1; i <= n;i++)
    {
       f[i] = 0;//记录数组初始化
       dis[i] = s[1][i];//记录依附于‘1’点的最小的边，以‘1’为起点作为依附点
    }
    f[1] = 1;
    for(i = 1; i< n;i++)//可以替换成for(i = 2; i<= n;i++)，循环其他的依附点
    {
        int min = inf;
        int u;//选定一个点
        for(j = 1; j <= n;j++)//寻找依附于该点最短的边
        {
            if(!f[j]&&dis[j] < min)
            {
                min = dis[j];
                u = j;
            }
        }
        if(min == inf)
        {
            flag = 1;
            break;
        }
        f[u] = 1;
        sum += min;
        for(k = 1; k<= n;k++)
        {
            if(f[k] == 0 && dis[k] > s[u][k])//已经记录'u'点，重新选择具有最小边的点
            {
                dis[k] = s[u][k];
            }
        }
    }
}
int main()
{
    int n,m,a,b,c;
    int i,j;
    while(scanf("%d%d",&n,&m)!= EOF)
    {
        sum = 0;
        flag = 0;
        memset(f,0,sizeof(f));
        for(i = 0 ;i <= n;i++)
        {
           for(j = 0; j<= n;j++)
           {
               if(i == j)
                s[i][j] = 0;
                else
                    s[i][j] = inf;
           }
        }
        for(i = 0; i< m;i++)
        {
            scanf("%d %d %d",&a,&b,&c);
            if(s[a][b] > c)//避免覆盖小值
            {
                s[a][b] = s[b][a] = c;
            }
        }
        prim(n);
        if(flag == 0)
            printf("%d\n",sum);
    }
    return 0;
}