还是畅通工程 HDU - 1233（最小生成树）

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/Miranda_ymz/article/details/86592909

还是畅通工程 HDU - 1233  

题意：中文题意就不用说了。

思路：很简单的模板题。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
//最小生成树 
const int maxn=100+10;//最大点数
const int maxm=maxn*maxn;
int a[maxn];
int f[maxn];//并查集使用
struct Edge{
	int u,v,w;
	bool operator <(const Edge &u)const{
		return w<u.w;
	}
}edge[maxm];
int tot;
void addedge(int u,int v,int w)
{
	edge[tot].u=u;
	edge[tot].v=v;
	edge[tot++].w=w;
} 
int find(int x)
{
	if(f[x]==-1) return x;
	return f[x]=find(f[x]);
}
int kruskal(int n)
{
	memset(f,-1,sizeof(f));
	sort(edge,edge+tot);
	int cnt=0;
	int ans=0;
	for(int i=0;i<tot;i++)
	{
		int u=edge[i].u;
		int v=edge[i].v;
		int w=edge[i].w;
		int t1=find(u);
		int t2=find(v);
		if(t1!=t2)
		{
			ans+=w;
			f[t1]=t2;
			cnt++;
		}
		if(cnt==n-1) break;
	}
	if(cnt<n-1) return -1;
	else return ans; 
}
int main() 
{
	int n,a,b,w;
	while(scanf("%d",&n)!=EOF&&n)
	{
		tot=0;
		int m=n*(n-1)/2;
		for(int i=0;i<m;i++){
			scanf("%d%d%d",&a,&b,&w);
			addedge(a,b,w);
			addedge(b,a,w);	
		}
		int ans=kruskal(n);
		printf("%d\n",ans);
	}
	return 0;
}