标签:
3 3 1 2 1 1 3 2 2 3 4 1 3 2 3 2 0 100
3 ?
#include<stdio.h> #include<stdlib.h> #include<string.h> #define INF 0x3f3f3f //定义 INF为无穷大 int n,m; int map[101][101]; //两点间距离 int vis[101]; //标记点是否存进生成树 int pay[101]; //记录距离并逐个比较没有标记的点,得到当前到"生成树(已被标记的点所在)"的最短距离 void prim() { int i,j,k,min,sum; memset(vis, 0, sizeof(vis));//清 0 防止下次循环累加或其他错误 for(i = 1; i <= m; i++) pay[i] = map[1][i];//1到其他所有点的距离 vis[1] = 1; //标记1,表示 1已经是 生成树的一个点 sum = 0; for(int i = 1; i < m; i++)//控制到 m-1 次 { min = INF; k = 1; for(j = 1; j <= m; j++) if(!vis[j] && pay[j] < min) { min = pay[j]; k = j;// k为标记满足条件的点的下标 } if(min == INF)//不满足条件 输出oh! { printf("?\n"); return; } vis[k] = 1; //标记过的点放进树 sum += min; //求出当前最短距离 for(j = 1;j <= m; j++)//更新 其余点到生成树的距离 { if( !vis[j] && pay[j] > map[k][j] )//如果j未在生成树内并且,当前保存的j到树的距离大于新更新的值 pay[j] = map[k][j]; } } printf("%d\n",sum); } int main() { int i,j,k; int a,b,c;//两个村庄 和成本 while(scanf("%d%d",&n,&m)!=EOF,n) { for(i = 1;i <= m;i++) { for(j = 1;j <= m; j++) { if(i == j)//先定义 点到点 自身到自身 距离==0 map[i][j] = 0; else // 任意两点间 距离为 无穷大! map[i][j] = INF; } } for( i = 0;i < n; i++) { scanf("%d%d%d",&a,&b,&c);//输入 两点 和边 map[a][b] = map[b][a] = c; //两点之间距离一样 1到2的距离 == 2到1距离 } prim(); } return 0; }
版权声明:原创文章,有借鉴之处,多多支持。hhh
hdoj 1863 畅通工程 【最小生成树 prim 算法】
标签:
原文地址:http://blog.csdn.net/liu6886/article/details/47441109