解题思路:最小生成树裸题,由于边的数目比较少,所以选用复杂度为O(ElogV)的kruskal算法,和最小生成树的区别是边的排序由大到小排。
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
struct edge
{
int u,v,w;
}s[50009];
int fa[20009],n,m,r;
bool cmp(edge a,edge b)
{
return a.w>b.w;
}
int find(int x)
{
if(x!=fa[x])
fa[x]=find(fa[x]);
return fa[x];
}
void unite(int x,int y)
{
int fax,fay;
fax=find(x);
fay=find(y);
fa[fax]=fay;
}
void kruskal()
{
int fax,fay,ans=0;
for(int i=0;i<r;i++)
{
fax=find(s[i].u);
fay=find(s[i].v);
if(fax!=fay)
{
ans+=s[i].w;
}
unite(s[i].u,s[i].v);
}
printf("%d\n",10000*(n+m)-ans);
}
int main()
{
int T,a,b,c;
//freopen("t.txt","r",stdin);
scanf("%d",&T);
while(T--)
{
scanf("%d%d%d",&n,&m,&r);
for(int i=0;i<r;i++)
{
scanf("%d%d%d",&a,&b,&c);
s[i].u=a;
s[i].v=b+n;
s[i].w=c;
}
sort(s,s+r,cmp);
for(int i=0;i<n+m;i++)
fa[i]=i;
kruskal();
}
return 0;
}