算法与数据结构第十次作业 征兵

这

这其实是求最小生成树的问题，用到了kurskal算法。这题应该是求最大生成树，然后再减一下。贴代码

#include <iostream>
#include <cstdio>
#include <algorithm>
#define MAX_E 50000
#define MAX_V 20000
#define COST 10000
using namespace std;
struct Edge{//定义边结构
	int From;
	int To;
	int Cost;
	void set(int f,int t,int c,int n){//为了区分男女，把女生的编号加N
		From=f;
		To=t+n;
		Cost=c;
	}
	bool operator >(Edge &t){//重载了大于号运算符
		return Cost>t.Cost;
	}
}e[MAX_E];
int v[MAX_V];
bool cmp(Edge x,Edge y){
	return x.Cost>y.Cost;
}

int Find(int x){//并查集的查找函数
	return x==v[x]?x:v[x]=Find(v[x]);
}

bool Union(int x,int y){//并查集的合并函数，用于判断是否已经是连通的
	int vx=Find(x);
	int vy=Find(y);
	if(vx==vy)
		return false;
	v[vx]=vy;
	return true;
}

void init_union_find(int vert){//初始化数组
	int i;
	for(i=0;i<vert;i++)
		v[i]=i;
}

int Kruskal(int vert,int e_num){
	int res=0;
	init_union_find(vert);
	sort(e,e+e_num,cmp);
	for(int i=0;i<e_num;i++){
		if(Union(e[i].From,e[i].To))
			res+=e[i].Cost;
	}
	return res;
}

int main()
{
	int N,M,R,i,x,y,d,ans;
	scanf("%d%d%d",&N,&M,&R);
	for(i=0;i<R;i++){
		scanf("%d%d%d",&x,&y,&d);
		e[i].set(x,y,d,N);
	}
	ans=Kruskal(N+M,R);
	ans=COST*(N+M)-ans;
	printf("%d\n",ans);
	return 0;
}

/*
5 5 8 
4 3 6831 
1 3 4583 
0 0 6592 
0 1 3063 
3 3 4975 
1 3 2049 
4 2 2104 
2 2 781 
*/