码迷,mamicode.com
首页 > 其他好文 > 详细

北大ACM2377——Bad Cowtractors~~最小生成树

时间:2015-05-06 15:12:39      阅读:104      评论:0      收藏:0      [点我收藏+]

标签:acm   北大   kruskal   

这题,也是简单的最小生成树。只是求解的是费用最大,而且可能出现不能形成一棵树的情况就输出 -1.

水题,直接附代码:

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;

class data
{
public:
	int from, to, cost;
};
data Edge[20005];

int par[10005];
int N, M, num;

int cmp(const data& a, const data& b)   //从大到小排序
{
	return a.cost > b.cost;
}
int finds(int x)                        //并查集查找函数
{
	if(x == par[x])
		return x;
	else
		return par[x] = finds(par[x]);
}
void join(int x, int y)                 //并查集合并函数
{
	x = finds(x);
	y = finds(y);
	if(x != y)
		par[y] = x;
}
int Kruskal()                           //用Kruskal算法求解最大费用
{
	int i, ans = 0;
	for(i = 0; i <= N; i++)
		par[i] = i;
	sort(Edge, Edge + num, cmp);        //排序
	for(i = 0; i < num; i++)
	{
		data e = Edge[i];
		if(finds(e.from) != finds(e.to)) //判断是否属于一个并查集
		{
			join(e.from, e.to);
			ans += e.cost;
		}
	}
	return ans;
}
int main()
{
	int a, b, c;
	while(scanf("%d%d", &N, &M) != EOF)
	{
		num = 0;
		for(int i = 0; i < M; i++)
		{
			scanf("%d%d%d", &a, &b, &c);
			Edge[num].from = a;
			Edge[num].to = b;
			Edge[num++].cost = c;
		}
		int ans = Kruskal();
		int flag = 0;
		for(int j = 1; j <= N; j++)         //判断是否不能形成一棵树
		{
			if(par[j] == j)
				flag++;
		}
		if(flag > 1)                       //不能,输出-1
			ans = -1;
		printf("%d\n", ans);
	}
	return 0;
}


北大ACM2377——Bad Cowtractors~~最小生成树

标签:acm   北大   kruskal   

原文地址:http://blog.csdn.net/qq_25425023/article/details/45534907

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!