Conscription POJ - 3723 （kruskal模板）

求一个最大生成树即可

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

const int N = 2e4 + 5, M = 5e4 + 5;
struct edge{
	int u, v, w;
	bool operator < (const edge& W )const {
		return w > W.w;
	}
}e[M];
int n, m, x, y, d, r, p[N];

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

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

int kruskal() {
	init();
	sort(e, e + r);
	int ans = 0;
	for(int i = 0; i < r; i++) {
		int fx = find(e[i].u);
		int fy = find(e[i].v);
		if(fx != fy) {
			p[fx] = fy;
			ans =ans + e[i].w;
		}
	}
	return ans;
}

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", &x, &y, &d);
			e[i].u = x;
			e[i].v = n + y;
			e[i].w = d;
		}
		int res;
		res = (n + m) * 10000 - kruskal();
		printf("%d\n", res);
	}
	return 0;
}