数据结构实验之图论九:最小生成树
Time Limit: 1000 ms Memory Limit: 65536 KiB
Problem Description
有n个城市,其中有些城市之间可以修建公路,修建不
同的公路费用是不同的。现在我们想知道,最少花多少钱修公路可以将所有的城市连在一起,使在任意一城市出发,可以到达其他任意的城市。
Input
输入包含多组数据,格式如下。
第一行包括两个整数n m,代表城市个数和可以修建的公路个数。(n <= 100, m <=10000)
剩下m行每行3个正整数a b c,代表城市a 和城市b之间可以修建一条公路,代价为c。
Output
每组输出占一行,仅输出最小花费。
Sample Input
3 2 1 2 1 1 3 1 1 0
Sample Output
2 0
Hint
Source
赵利强
#include<stdio.h>
#include<string.h>
const int inf=0x3f3f3f;
const int maxn=1010;
int n,m;
int mp[maxn][maxn];
int vis[maxn];
int b[maxn];
int prim()
{
int sum=0;
int i,j,v,ma;
vis[1]=1;
for(i=1; i<=n; i++)
b[i]=mp[1][i];//记录与第一个节点连接的权值
for(i=1; i<n; i++)//还需要n-1个顶点
{
ma=inf;
for(j=1; j<=n; j++)
{
if(vis[j]==0&&ma>b[j])
{
ma=b[j];
v=j;
}
}
if(ma==inf)
return 0;
sum+=ma;
vis[v]=1;
for(j=1; j<=n; j++)
{
if(vis[j]==0&&b[j]>mp[v][j])
b[j]=mp[v][j];
}//更新还没有加入的点的距离
}
return sum;
}
int main()
{
while(scanf("%d%d",&n,&m)!=EOF)
{
int i,j;
memset(vis,0,sizeof(vis));
for(i=1; i<=n; i++)
{
for(j=1; j<=n; j++)
{
if(i==j)
mp[i][j]=0;//对角线的权值置为0
else mp[i][j]=inf;//其他无穷大常量
}
}
for(i=0; i<m; i++)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
if(mp[a][b]>c)
mp[a][b]=mp[b][a]=c;
}
printf("%d\n",prim());
}
return 0 ;
}