Description
某省调查城镇交通状况,得到现有城镇道路统计表,表中列出了连接两个城镇需要花费的代价。省政府“畅通工程”的目标是使全省任何两个城镇间都可以实现交通(但不一定有直接的道路相连,只要互相间接通过道路可达即可)。问最少花费多少代价就可以完成工程?
Input
输入包含多组数据,对于每组测试数据
第一行包含两个正整数N和M(0 <=N <=1000,0 < M < 5000),分别代表现有城镇的数目和已修建的道路的数目。城镇分别以1~N编号。
接下来是M行道路信息。每一行有三个整数A,B,X(0 < A,B <= N ,0 < X < 10000),表示可以在城镇A和城镇B之间建一条花费为X的双向道路。保证数据可以完成工程。
Output
对于每组数据,输出完成工程需要花费的最少代价。
Sample Input
3 4
1 2 3
2 1 4
1 3 1
2 3 2
Sample Output
3
这个算法总体上比prim算法好用,如何构建我就不赘述了,就是加边法,依次找最小的边,
再看看找到的边是否已经属于一个集合了,如果不是就ans+=wei,并且把他放到集合里
#include<iostream>
#include<algorithm>
using namespace std;
struct node
{
int fa;
int son;
int wei;
bool operator < (const node& a)const
{
return wei<a.wei;
}
}edge[100005];
int father[100005];
int finds(int x)
{
while(father[x]!=x)
x=father[x];
return x;
}
int main()
{
int n,m,ans;
while(cin>>n>>m)
{
ans=0;
for(int i=1;i<=n;i++)
father[i]=i;
for(int i=1;i<=m;i++)
cin>>edge[i].fa>>edge[i].son>>edge[i].wei;
sort(edge+1,edge+1+m);
for(int i=1;i<=m;i++)
{
int fa=edge[i].fa;
int son=edge[i].son;
if(finds(fa)!=finds(son))
{
ans+=edge[i].wei;
father[finds(fa)]=father[finds(son)];
}
}
cout<<ans<<endl;
}
return 0;
}