POJ 3723 Conscription G++ kruskal实现 prim未实现

#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
//英语 最小生成树   kruskal实现   prim未实现
struct nod{
	int x;
	int y;
	int zhi;
};
bool cmp(nod a,nod b)
{
	return a.zhi<b.zhi;
}
int fa[20008];
int find(int x)
{
	if(fa[x]==-1)
	{
		return x;
	}else
	{
		return fa[x]=find(fa[x]);
	}
}
int main()
{
	int T;
	scanf("%d",&T);
	for(int o=0;o<T;o++)
	{
		memset(fa,-1,sizeof(fa));
		int n,m,k;
		//cin>>n>>m>>k;
		scanf("%d%d%d",&n,&m,&k);
		vector<nod> ve;
		for(int i=0;i<k;i++)
		{
			nod t;
			scanf("%d%d%d",&t.x,&t.y,&t.zhi); 
			//cin>>t.x>>t.y>>t.zhi;
			t.y=t.y+n;
			t.zhi=10000-t.zhi;
			ve.push_back(t);
		}
		sort(ve.begin(),ve.end(),cmp);
		int sum=0;
		int jl=0;
		for(int i=0;i<ve.size();i++)
		{
			int tx=ve[i].x;
			int ty=ve[i].y;
			int tz=ve[i].zhi;
			if(find(tx)!=find(ty))
			{
				jl=fa[find(tx)]=find(ty);
				sum=sum+tz;
				//cout<<tx<<" "<<ty<<" "<<tz<<endl;
				//cout<<sum<<endl;
			}
		}
		for(int i=0;i<(n+m);i++)
		{
			if(find(i)!=jl)
			{
				fa[find(i)]=jl;
				//cout<<i<<endl;
				sum=sum+10000;
			}
		}
		cout<<sum+10000<<endl;
	} 
	return 0;
}