POJ_3723_Conscription

最新推荐文章于 2021-03-08 06:08:38 发布

原创最新推荐文章于 2021-03-08 06:08:38 发布 · 349 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#poj #3723 #conscription

ACM 专栏收录该内容

78 篇文章

订阅专栏

本文介绍了一种使用并查集实现Kruskal算法的方法，用于解决最小生成树问题。通过具体代码展示了如何初始化并查集、查找根节点、合并集合以及判断两个元素是否属于同一集合等操作。

//见《挑战程序设计竞赛》第2.5.6节

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

#define MAX_N 20000
#define MAX_E 50000

int par[MAX_N];
int ran[MAX_N];

void init(int n) {
	for (int i = 0; i < n; i++) {
		par[i] = i;
		ran[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 (ran[x] < ran[y])
		par[x] = y;
	else {
		par[y] = x;
		if (ran[x] == ran[y]) ran[x]++;
	}
}

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

struct edge { int u, v, cost; }es[MAX_E];
bool compare(const edge& e1, const edge& e2) { return e1.cost > e2.cost; };

int main() {
	int T, N, M, R, save, all;
	scanf("%d", &T);
	while (T--) {
		scanf("%d%d%d", &N, &M, &R);
		all = N + M;
		for (int i = 0; i < R; i++) {
			scanf("%d%d%d", &es[i].u, &es[i].v, &es[i].cost);
			es[i].v += N;
		}
		sort(es, es + R, compare);
		init(all);
		save = 0;
		for (int i = 0; i < R; i++) {
			if (!same(es[i].u, es[i].v)) {
				unite(es[i].u, es[i].v);
				save += es[i].cost;
			}
		}
		printf("%d\n", (N + M) * 10000 - save);
	}
}