最小生成树

题目水过一遍，练练手

 

 

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>

using namespace std;

const int inf=999999;
int map[100][100];
int dis[100];
int vis[100];
int sum;

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

int main()
{
    int i,j;
    int n,m;
    while(cin>>n>>m)
    {
        sum=0;
        for(i=1; i<=n; i++)
        {
            for(j=1; j<=n; j++)
            {
                map[i][j]=inf;
            }
            map[i][i]=0;
        }
        int x,y,z;
        for(i=1; i<=m; i++)
        {
            cin>>x>>y>>z;
            if(map[x][y]>z)
            {
                map[x][y]=map[y][x]=z;
            }
        }
        prim(n);
        cout<<sum<<endl;
    }
    return 0;
}