3362=数据结构实验之图论六：村村通公路

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#define INF 99999999
int map[1000][1000],dis[1000],visited[10000];
int sum;
void prim(int n)
{
    int count=1,i,j,k;
    while(count<=n)
    {
        count++;
        int min=INF;
        for(i=1; i<=n; i++)
        {
            if(visited[i]==0&&dis[i]<min)
            {
                min=dis[i];
                j=i;
            }
        }
        if(min==INF)
        {
            sum=-1;
            return;
        }
        visited[j]=1;
        sum+=dis[j];
        for(k=1; k<=n; k++)
        {
            if(visited[k]==0&&dis[k]>map[j][k])
                dis[k]=map[j][k];
        }
    }

}
int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        int i,j;
        sum=0;
        memset(visited,0,sizeof(visited));
        for(i=1; i<=n; i++)
        {
            for(j=1; j<=n; j++)
            {
                if(i==j)
                    map[i][j]=0;
                else
                    map[i][j]=INF;
            }
        }
        while(m--)
        {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            map[a][b]=map[b][a]=c;
        }
        for(i=1; i<=n; i++)
        {
            dis[i]=map[1][i];
        }
        prim(n);
        printf("%d
",sum);
    }
    return 0;
}