HDU1233 还是畅通工程

畅通工程系列,中文题目就不解释意思了。

题目比较直白,直接最小生成树。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<functional>
#include<algorithm>
using namespace std;
struct node
{
	int from,to;
	int cost;
}e[100*100+10];
int fa[105];
int n,m,i,sum;
int cmp(struct node a,struct node b){return a.cost<b.cost;}
void init()
{
	sum=0;
	for(i=1;i<=n;i++)
		fa[i]=i;
}
int getfa(int x)
{
	if(x==fa[x])
		return x;
	else
		return fa[x]=getfa(fa[x]);
	//return x==fa[x]?x:fa[x]=getfa(fa[x]);
}
int merge(int u,int v)
{
	int t1,t2;
	t1=getfa(u);
	t2=getfa(v);
	//printf("u=%d v=%d t1=%d t2=%d\n",u,v,t1,t2);
	if(t1!=t2)
	{
		fa[t1]=t2;
		return 1;
	}
	return 0;
}
int main()
{
	int t1,t2,t3;
	while(scanf("%d",&n),n)
	{
		init();
		m=n*(n-1)/2;
		for(i=1;i<=m;i++)
		{
			scanf("%d%d%d",&t1,&t2,&t3);
			e[i].from=t1;
			e[i].to=t2;
			e[i].cost=t3;
		}
		sort(e+1,e+m+1,cmp);
		for(i=1;i<=m;i++)
		{
			if(merge(e[i].from,e[i].to))
			{
				sum+=e[i].cost;
				//printf("cost=%d\n",e[i].cost);
			}
		}
		printf("%d\n",sum);
	}
}


全部评论

相关推荐

11-09 01:22
已编辑
东南大学 Java
高级特工穿山甲:羡慕,我秋招有家企业在茶馆组织线下面试,约我过去“喝茶详谈”😢结果我去了发现原来是人家喝茶我看着
点赞 评论 收藏
分享
过往烟沉:我说什么来着,java就业面就是广!
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务