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

数据结构实验之图论九：最小生成树
Time Limit: 1000 ms Memory Limit: 65536 KiB

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

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int MAXV=1010;
const int INF=0x3f3f3f3f;
int n;
int bian[MAXV][MAXV],d[MAXV],vis[MAXV];///分别记录权值，到防护罩的距离，访问位状态
int prim()
{
    //memset(d,INF,sizeof(d));
    fill(d,d+MAXV,INF);
    d[1]=0;
    int ans=0;
    for(int i=1;i<=n;i++)
    {
        int u=-1,min=INF;
        for(int j=1;j<=n;j++)
        {
            if(vis[j]==0&&d[j]<min)
            {
                u=j;
                min=d[j];
            }
        }
            vis[u]=1;
            ans+=d[u];
        ///优化与u相连的未访问的点v到集合的最短距离
            for(int v=1;v<=n;v++)
            {
                if(vis[v]==0&&bian[u][v]!=INF&&bian[u][v]<d[v])
                {
                    d[v]=bian[u][v];
                }
         }
    }
    return ans;
}
int main()
{
    int m,v,u,c;
    while(cin>>n>>m)
    {
        fill(bian[0],bian[0]+MAXV*MAXV,INF);
        fill(vis,vis+MAXV,0);
        //memset(vis,0,sizeof(vis));
        for(int i=0;i<m;i++)
        {
            cin>>v>>u>>c;
              if(bian[v][u]>c) ///此处不加判断提交报错，
                bian[v][u]= bian[u][v]= c;

        }
        int sum=prim();
        printf("%d\n",sum);
        //else printf("-1\n");
    }
}