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

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

Time Limit: 1000MS Memory Limit: 65536KB

Submit 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 <cstring>
using namespace std;
#define maxi 0x3f3f3f3f
int n,m,mp[1100][1100],vis[1100],k;
int minp[1100]; //记录最短距离
int prim();
int main()
{
        int u,v,p;
    while(cin>>n>>m)
    {
            memset(mp,0,sizeof(mp));
            memset(vis,0,sizeof(vis));
            memset(minp,0,sizeof(minp));
           for(int i=1;i<=n;i++)
           {
                 for(int j=1;j<=n;j++)
                 {
                        if(i==j)
                        mp[i][j]=0;
                        else
                                mp[i][j]=maxi;
                 }
           }
           for(int i=0;i<m;i++)
           {
                   cin>>u>>v>>p;
                   if(p<mp[u][v])
                        mp[u][v]=mp[v][u]=p;
           }
           k=0;
           int sum=prim();
           if(k==n-1)
                cout<<sum<<endl;
           else
                cout<<-1<<endl;
    }
}
int prim()
{
        int sum=0;
       for(int i=1;i<=n;i++)
       {
               minp[i]=mp[1][i];  //注意此时i是下一个可能节点编号
       }
       vis[1]=1;    //初始化第一个点
       for(int i=1;i<=n;i++)
       {
               int m=maxi,pos; //初始化m 记录
               for(int j=1;j<=n;j++)
               {
                       if(vis[j]!=1&&m>minp[j])  //没有遍历这条边,并且
                       {
                               m=minp[j];
                               pos=j;
                       }
               }
               if(m==maxi)
                break;

               vis[pos]=1;
               sum=sum+m;
               k++;

               for(int i=1;i<=n;i++)
               {
                       if(vis[i]!=1&&mp[pos][i]<minp[i])
                       {
                               minp[i]=mp[pos][i];
                       }
               }//新得最小值，更新
       }
       return sum;
}

//理解：相当于一棵知道一个节点找下一个节点的树，找这个节点离得最近的点就是下一个节点，
//依次进行,直到找到n-1个下一个节点，结束