题目描述
当前农村公路建设正如火如荼的展开,某乡镇政府决定实现村村通公路,工程师现有各个村落之间的原始道路统计数据表,表中列出了各村之间可以建设公路的若干条道路的成本,你的任务是根据给出的数据表,求使得每个村都有公路连通所需要的最低成本。
输入
连续多组数据输入,每组数据包括村落数目N(N <= 1000)和可供选择的道路数目M(M <= 3000),随后M行对应M条道路,每行给出3个正整数,分别是该条道路直接连通的两个村庄的编号和修建该道路的预算成本,村庄从1~N编号。
输出
输出使每个村庄都有公路连通所需要的最低成本,如果输入数据不能使所有村庄畅通,则输出-1,表示有些村庄之间没有路连通。
示例输入
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
示例输出
19
#include <iostream> #include <algorithm> #include <cstring> #include <cstdio> using namespace std; #define inf 999999999 int map1[1024][1024]; int vis[1024]; int dist[1024]; int n,m; void prime() { int minn; int i,j,v; int sum=0; for(i=1;i<=n;i++) dist[i]=map1[1][i]; vis[1]=1; for(i=2;i<=n;i++) { minn=inf; v=-1; for(j=1;j<=n;j++) { if(dist[j]<minn&&!vis[j]) {minn=dist[j]; v=j; } } if(v!=-1) { sum=sum+dist[v]; vis[v]=1; for(j=1;j<=n;j++) { if(!vis[j]&&map1[v][j]<dist[j]) dist[j]=map1[v][j]; } } } for(i=1;i<=n;i++) { if(!vis[i]) { sum=-1; break; } } cout<<sum<<endl; } int main() { int u,v,w,i,j; while(scanf("%d %d",&n,&m)!=EOF) { for(i=1;i<=n;i++) for(j=1;j<=n;j++) { if(i==j) map1[i][j]=map1[j][i]=0; else map1[i][j]=map1[j][i]=inf; } memset(vis,0,sizeof(vis)); for(i=0;i<m;i++) { cin>>u>>v>>w; map1[u][v]=map1[v][u]=w; } prime(); } }