数据结构实验之图论九:最小生成树
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
注:本题采用的prim方法,先将一个点标记,用它更新下方的edge结构体数组,
找出未标记点与所标记过的点相连接的权最小的边,
用该边相连的未标记的点刷新edge数组,直到所有的点都完成遍历。
#include <bits/stdc++.h>
#include <iostream>
using namespace std;
//可能输入时更新数据
#define inf 0x3f3f3f3f
int rode[102][102];
int vis[102];
struct minedge
{
int bian;
int value;
};
int main()
{
int n,m,a,b,c,i,t,sum,num,k;
struct minedge edge[102];
while(cin>>n>>m)
{
memset(rode,inf,sizeof(rode));///数组初始化
memset(vis,0,sizeof(vis));
for(i=1; i<=m; i++)
{
cin>>a>>b>>c;
///可能会发生相同的权更新边,本人因此曾改了两个小时的bug,
///感谢那位坐在我旁边的伙伴帮我找出错误
if(c<rode[a][b])
{
rode[a][b]=c;
rode[b][a]=c;
}
}
for(i=1; i<102; i++)
{
edge[i].bian=0;
edge[i].value=inf;
}
t=0;
k=1;
edge[1].bian=1;
vis[1]=1;//将第一个点标记
while(t<n)
{
for(i=1; i<=n; i++)
{
if(rode[k][i]!=inf&&rode[k][i]<edge[i].value&&vis[i]!=1)
{
edge[i].bian=i;
edge[i].value=rode[k][i];
}
}
num=inf;
for(i=1; i<=n; i++)
{
if(edge[i].value<num&&vis[i]==0)
{
num=edge[i].value;
k=edge[i].bian;
}
}
vis[k]=1;
t++;
}
sum=0;
for(i=1; i<=n; i++)
{
if(edge[i].value!=inf) sum+=edge[i].value;
}
t=0;
for(i=1;i<=n;i++)
{
if(edge[i].bian==0)break;
else t++;
}
if(t==n)cout<<sum<<endl;
else cout<<"0"<<endl;
}
return 0;
}
/***************************************************
Result: Accepted
Take time: 44ms
Take Memory: 240KB
****************************************************/
注意理解,自己理解才是王道!!!