POJ 3723 Conscription MST

http://poj.org/problem?id=3723

题目大意:

需要征募女兵N人,男兵M人,没征募一个人需要花费10000美元,但是如果已经征募的人中有一些关系亲密的人,那么可以少花一些钱,给出若干男女之间的1~9999之间的亲密度关系,征募某个人的费用是10000-(已经征募的人中和自己的亲密度的最大值),要求通过适当的征募顺序使得征募所有人的花费最小。

思路:

好几天没刷题,来水一发

征募了某个人a时,如果使用了a和b之间的关系,那么就连一条从a到b的边。那么根据贪心算法,每次选择最大的边,那么最后减去的花费就最多,也就是求最大生成树。



#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int cost=10000;
const int MAXN=20000+10;
const int MAXM=50000+10;
int n,m,r,fa[MAXN];
struct edge
{
	int from,to,val;
}e[MAXM];

bool operator <(const edge& a,const edge &b)
{
	return a.val>b.val;
}

int find(int cur)
{
	return cur==fa[cur]? cur:fa[cur]=find(fa[cur]);
}

int main()
{
	int T;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d%d%d",&n,&m,&r);
		for(int i=0;i<r;i++)
		{
			scanf("%d%d%d",&e[i].from,&e[i].to,&e[i].val);
			e[i].to+=n;
		}
		
		for(int i=0;i<=n+m;i++)
			fa[i]=i;
		sort(e,e+r);
		int sum=0;
		for(int i=0;i<r;i++)
		{
			int x=e[i].from,y=e[i].to;
			int root_x=find(x),root_y=find(y);
			if(root_x==root_y) continue;
			fa[root_x]=root_y;
			sum+=e[i].val;
		}
		printf("%d\n",(n+m)*cost-sum);
	}
	return 0;
}


posted @ 2014-03-04 23:11  hr_whisper  阅读(142)  评论(0编辑  收藏  举报