poj 2377最大生成树 Kruskal

点击打开链接

#include <iostream>
#include <algorithm>
using namespace std;
const int M = 21010;
struct edge{
	int u;
	int v;
	long long cost;
	bool operator <(const edge &t)const
	{
		return cost>t.cost; 
	}
}e[M];
int n,m,fa[M];
void Inin()
{
	for(int i=1;i<=n;i++)
	{
		fa[i]=i;
	}
}
int find(int x)
{
	if(x!=fa[x])
	{
		fa[x]=find(fa[x]); //路径压缩		
	}
	return fa[x];
	
}
void Union(int x,int y)
{
	int a=find(x);
	int b=find(y);
	if(a!=b)
	{
		fa[a]=b;
	}
}
void Kruskal()
{
	Inin();// union-set 初始化 
	long long ans=0;
	for(int i=0;i<m;i++)
	{
		if(find(e[i].u)!=find(e[i].v))
		{
			Union(e[i].u,e[i].v);
			ans+=e[i].cost; 
			n--;
		}
		if(n==1)
		break;
	}
	if(n==1)
	cout<<ans<<endl;
	else// 连不通 
	cout<<-1<<endl;
}
int main() 
{
	cin>>n>>m;
	for(int i=0;i<m;i++)
	{
		cin>>e[i].u>>e[i].v>>e[i].cost;
	}
	sort(e,e+m);
	Kruskal();//将n个连通分量变减少至1个(最小生成树)时 所用花费最少 
	return 0;
}