5-10 公路村村通

5-10 公路村村通   (30分)

现有村落间道路的统计数据表中，列出了有可能建设成标准公路的若干条道路的成本，求使每个村落都有公路连通所需要的最低成本。

输入格式:

输入数据包括城镇数目正整数$N$（$\le 1000$）和候选道路数目$M$（$\le 3N$）；随后的$M$行对应$M$条道路，每行给出3个正整数，分别是该条道路直接连通的两个城镇的编号以及该道路改建的预算成本。为简单起见，城镇从1到$N$编号。

输出格式:

输出村村通需要的最低成本。如果输入数据不足以保证畅通，则输出$-1$，表示需要建设更多公路。

输入样例:

6 15
1 2 5
1 3 3
1 4 7
1 5 4
1 6 2
2 3 4
2 4 6
2 5 2
2 6 6
3 4 6
3 5 1
3 6 1
4 5 10
4 6 8
5 6 3

输出样例:

12

最小生成树Kruskal算法模板题。

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=1010;
const int M=N*3;
struct Edge{
	int u,v,co;
}edge[M];
int n,m;
int fa[N];

bool cmp(const Edge &a,const Edge &b)
{
	return a.co<b.co;
}
int find(int x)
{
	return fa[x]<0?x:fa[x]=find(fa[x]);
}
bool join(int x,int y)
{
	int fx=find(x);
	int fy=find(y);
	if(fx!=fy)
	{
		fa[fx]=fy;
		return true;
	}
	return false;
}
long long kruskal()
{
	int cnt=0;
	long long sum=0;
	for(int i=1;i<=m;i++)
	{
		if(join(edge[i].u,edge[i].v))
		{
			sum+=edge[i].co;
			cnt++;
		}
		if(cnt==n-1)
			break;
	}
	if(cnt<n-1)
		return -1;
	return sum;
}
int main()
{
	cin>>n>>m;
	memset(fa,-1,sizeof(fa));
	for(int i=1;i<=m;i++)
	{
		cin>>edge[i].u>>edge[i].v>>edge[i].co;
	}
	sort(edge+1,edge+m+1,cmp);
	cout<<kruskal()<<endl;
	return 0;
}