标签:
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