poj3723 Conscription 最小生成树

原题链接：poj3723 Conscription 

#include <cstdio>
#include <cstring> 
#include <algorithm>
using namespace std;
const int MAX_N = 10010 * 2;
const int MAX_E = 50010;
int N, M, V, E;		//顶点数和边数
struct edge{	//cost为u到v边的长度 
	int u, v, cost;
};
edge es[MAX_E];
bool comp(const edge& e1, const edge& e2){
	return e1.cost < e2.cost;
}
int par[MAX_N], rank[MAX_N];//父亲,树的高度 
void init_union_find(int n){//初始化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和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){//判断x和y是否属于同一个集合 
	return find(x) == find(y);
}
int kruskal(){
	sort(es, es + E, comp);	//按照edge.cost的顺序从小到大排列 
	init_union_find(V);						//并查集的初始化 
	int res = 0;
	for(int i = 0;i < E;i ++){
		edge e = es[i];
		if(!same(e.u, e.v)) {
			unite(e.u, e.v);
			res += e.cost;
		}
	}
	return res;
} 
int main(){
	int T,R, x, y, d;
	int i;
	scanf("%d", &T);
	while(T --){
		scanf("%d%d%d", &N, &M, &E);
		V = N+M;
		for(i = 0;i < E;i ++){
			scanf("%d%d%d", &x, &y, &d);	
			es[i].u = x;	es[i].v = N + y;	es[i].cost = -d; // es[i]= (edge){x, N + y, -d};
		}
		printf("%d\n", 10000 * V + kruskal());
	}
	return 0;
}