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

Statistic
Problem Description

当前农村公路建设正如火如荼的展开，某乡镇政府决定实现村村通公路，工程师现有各个村落之间的原始道路统计数据表，表中列出了各村之间可以建设公路的若干条道路的成本，你的任务是根据给出的数据表，求使得每个村都有公路连通所需要的最低成本。
Input

连续多组数据输入，每组数据包括村落数目N(N <= 1000)和可供选择的道路数目M(M <= 3000)，随后M行对应M条道路，每行给出3个正整数，分别是该条道路直接连通的两个村庄的编号和修建该道路的预算成本，村庄从1～N编号。
Output

输出使每个村庄都有公路连通所需要的最低成本，如果输入数据不能使所有村庄畅通，则输出-1，表示有些村庄之间没有路连通。
Example Input

5 8
1 2 12
1 3 9
1 4 11
1 5 3
2 3 6
2 4 9
3 4 4
4 5 6
Example Output

19

#include <iostream>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#define max 1010011
using namespace std;
int n, m;
int map[1010][1010], vis[1010];
int dis[1010];
int sum;
void prim()
{
   sum=0;
   int i, j;
   for(i=1; i<=n; i++)
   {
       dis[i] = map[1][i];
   }
   vis[1] = 1;//找到第一个点，没有权值

   int pos;
   int flag=0;
   for(i=1; i<n; i++)
   {
       int mm = max;//用mm保存最小的距离
       for(j=1; j<=n; j++)
       {
           if(vis[j]==0&&dis[j] < mm)
           {
               mm = dis[j];
               pos = j;
           }
       }

       if(mm==max)
       {
           flag=1;
           break;
       }
       vis[pos] = 1;
       sum += mm;//加上距离最近的点的权值
       for(j=1; j<=n; j++)//更新dis,如果map存在更小的，
       {
           if(vis[j]==0&&dis[j] > map[pos][j])
           {
               dis[j] = map[pos][j];
           }
       }
   }
   if(flag==0)
    printf("%d\n", sum);
   else
   printf("-1\n");
}
int main()
{
    while(scanf("%d %d", &n, &m)!=EOF)
    {
        memset(map, max, sizeof(map));
        memset(vis, 0, sizeof(vis));
        int i;
        for(i=0; i<m; i++)
        {
            int x, y, z;
            cin>>x>>y>>z;
            map[x][y]=map[y][x]=z;
        }
        prim();
    }
    return 0;
}