數據結構實驗之圖論九：最小生成樹

#include <bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f3f //定義最大值
int ma[150][150],v[150],dist[150];
int visit[150];
int sum，flag;
void prime(int n)
{
    int i,j,k;
    for(i=1;i<=n;i++)
    {
        dist[i]=ma[1][i];
    }
    visit[1]=1;
    for(i=2;i<=n;i++)
    {
        int minn=INF;
        int u;
        for(j=1;j<=n;j++)
        {
            if(visit[j]==0&&dist[j]<minn)
            {
                minn=dist[j];
                u=j;
            }
        }
        if(minn==INF)
        {
          flag=1;
          break;
        }
        visit[u]=1;
        sum+=minn;
        for(k=1;k<=n;k++)
        {
            if(visit[k]==0&&dist[k]>ma[u][k]) // 就這裏和DISTL不一樣，這裏是更新了dist[]的起點，改爲u。
            {
                dist[k]=ma[u][k];
            }
        }
    }
}
int main()
{
     int n, m, a, b, c;
    while(~scanf("%d %d", &n, &m))
    {
        sum=0;
        flag=0;
        memset(visit,0,sizeof(visit));
        for(int i=0;i<=n;i++)//初始化矩陣
        {
            for(int j=0;j<=n;j++)
            {
                if(i==j)
                    ma[i][j] = 0;
                else
                    ma[i][j] = INF;
            }
        }
        for(int i=0;i<m;i++)
        {
            scanf("%d %d %d", &a, &b, &c);
            if(ma[a][b]>c)//避免覆蓋小值
            {
                ma[a][b] = ma[b][a] = c;
            }
        }
        prime(n);
        if(flag==0)
        {
          printf("%d\n",sum);
        }
    }
    return 0;
}