kruskal算法poj3723

题意:征兵,征一个兵要10000块钱,一些男人和女人的关系,如果有一个男的和一个女的,他们的关系值是d,那么如果他们的其中一个已经入伍了,另一个就只需要10000-d的钱就可以入伍,每个人入伍的时候只能用一个关系。求至少需要多少钱。

最多需要10000*人数的钱,需要利用他们之间的关系来减少,把他们之间的关系都变成负的,求一下最小生成树就好了

#pragma warning(disable:4996)
#include <cstring>
#include<cstdio>
#include <algorithm>
#include<iostream>
#include<cmath>
#include<queue>
#include<set>
#include<vector>
#include<stack>
using namespace std;
typedef long long ll;
const int inf = INT_MAX;
const int maxn = 10005;
int par[2*maxn], n, m, R;//数组要开够
//存储边
struct node
{
	int u, v, w;
};
node edge[5*maxn];
//按照边权值排序
bool cmp(node x, node y)
{
	return x.w < y.w;
}
//并查集
int find(int x)
{
	if (par[x] == x)return par[x];
	else return par[x] = find(par[x]);
}
void uni(int x, int y)
{
	int px, py;
	px = find(x);py = find(y);
	if (px < py)par[py] = px;
	else par[px] = py;
}
int kruskal()
{
	int sum = 0, i, j,num=0;
	for (i = 0;i < R;i++)
	{
		int u = edge[i].u, v = edge[i].v;
		if (find(u) != find(v))//如果没有联通
		{
			sum += edge[i].w;
			uni(u, v);
			num++;
		}
		if (num >= (n + m - 1))break;//形成了生成树
	}
	return sum;
}
int main()
{
	int T, i, j, t1, t2, t3;
	scanf("%d", &T);
	while (T--)
	{
		scanf("%d%d%d", &n, &m, &R);
		for (i = 0;i < n + m;i++)
		{
			par[i] = i;
		}
		for (i = 0;i < R;i++)
		{
			scanf("%d%d%d", &t1, &t2, &t3);
			edge[i].u = t1;edge[i].v = t2+n;//另一个顶点的编号要加n
			edge[i].w = (-t3);
		}
		sort(edge, edge + R, cmp);
		int ans = (m + n) * 10000;
		ans += kruskal();
		printf("%d\n", ans);
	}
	return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值