POJ 3723 Conscription (最小生成树 kruskal)

在POJ交题最好不用c++ 很容易卡时间

用c就过了

猛男落泪。

kruskal算法最小生成树

无向图求解最大权森林的问题

需要注意两个点:

1.男女序号问题 他们需要有唯一标识,这样才能正常连起来(男兵m编号+n就行了)。

2.cost权值需要变负,因为求得是男女之间最大的亲密度(要最小花费嘛)加个负号排序才能最大优先啊。

 /*
 qq:1239198605
 ctgu_yyf
        */

#include<iostream>
#include<cstdio>
#include<string>
#include<vector>
#include<queue>
#include<stack>
#include<cstring>
#include<algorithm>
#include<cmath>
#define ll long long
using namespace std;
struct Edge{
	int u,v,cost;
};

Edge edge[50005];
int e;//点的个数 
int n,m,pairr;//题目数据 
int par[20005];//父节点 
int rank[20005];//树的高度 //深度 也可以不要 

bool cmp(const Edge &a,const Edge &b)
{
	return a.cost<b.cost;
}

void init(int n)
{
	for(int i=0;i<n;i++)
    {
	par[i]=i;
	rank[i]=0;}
}

int find(int x)
{
	if(par[x]==x)
	return x;
	else
	return par[x]=find(par[x]);
}

void unite(int x,int y)
{
	x=find(x);
	y=find(y);
	if(x==y)
	return ;
	
	if(rank[x]<rank[y])
	par[x]=y;
	else
	{
		par[y]=x;
		if(rank[x]==rank[y])
		rank[x]++;
	}
}

bool same(int x,int y)
{
	return find(x)==find(y);
}

int kruskal()
{
	sort(edge,edge+pairr,cmp);
	
	
//	for(int i=0;i<pairr;i++)
//	cout<<"**"<<edge[i].u<<" "<<edge[i].v<<" "<<edge[i].cost<<endl;
	
	init(e);
	
	int res=0;
	for(int i=0;i<pairr;i++)
	{
		if(same(edge[i].u,edge[i].v)==false)
		{
		 unite(edge[i].u,edge[i].v);
		 res=res+edge[i].cost;
	    }
	}
	
	return res;
}

int main()
{
   ios::sync_with_stdio(false);
   int t;
   scanf("%d",&t);
   while(t--)
   {
   	scanf("%d%d%d",&n,&m,&pairr);
   	e=n+m;
   	
   	for(int i=0;i<pairr;i++)
   	{
   		scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].cost);
   		//cin>>edge[i].u>>edge[i].v>>edge[i].cost;
   		edge[i].v=edge[i].v+n;
   		edge[i].cost=-edge[i].cost;
	}
	
	
   	int ans=(n+m)*10000+kruskal();
   	printf("%d\n",ans);
   	
	} 


return 0;
}

猜你喜欢

转载自blog.csdn.net/k_koris/article/details/81563766