km算法

作用

求二分图最大权完美匹配

参考博客

KM算法详解+模板

代码实现

存储结构

const int maxn=310;
const int inf=0x3f3f3f3f;
int mmp[maxn][maxn]; // 记录每个妹子和每个男生的好感度
bool book_gril[maxn];//标记被匹配过的人
bool book_boy[maxn];
int val_gril[maxn];//权值
int val_boy[maxn];
int match[maxn];//记录匹配到的人
int slack[maxn];//尚需的期望值

所需函数

dfs递归匹配

bool dfs(int gril){
	book_gril[gril]=true;
	for(int boy=0;boy<n;boy++){
		if(book_boy[boy]) continue;// 每一轮匹配每个男生只尝试一次
		int x=val_gril[gril]+val_boy[boy]-mmp[gril][boy];//计算值是否匹配
		if(x==0){//如果匹配
			book_boy[boy]=true;
			if(match[boy]==-1||dfs(match[boy])){// 找到一个没有匹配的男生 或者该男生的妹子可以找到其他人
				match[boy]=gril;
				return true;
			}
			
		}
		else{
			slack[boy]=min(slack[boy],x); // slack 男生要得到女生的倾心还需多少期望值取最小
		}
	}
	return false;
}

km函数

int km(){
	memset(match,-1,sizeof(match));
	memset(val_boy,0,sizeof(val_boy));
	for(int i=0;i<n;i++){
		val_gril[i]=mmp[i][0];
		for(int j=1;j<n;j++){
			val_gril[i]=max(val_gril[i],mmp[i][j]);// 初始化每个女生期望值是与她相连的男生最大的好感度
		}
	}
	 // 尝试为每一个女生解决归宿问题
	for(int i=0;i<n;i++){
		fill(slack,slack+n,inf);// 因为要取最小值 初始化为无穷大
		while(1){
			memset(book_gril,false,sizeof(book_gril));
			memset(book_boy,false,sizeof(book_boy));
			if(dfs(i)) break;// 找到归宿 退出
			int v=inf;
			// 最小可降低的期望值
			for(int j=0;j<n;j++)if(!book_boy[j]) v=min(v,slack[j]);
			for(int j=0;j<n;j++){
				 // 所有访问过的女生降低期望值
				if(book_gril[j]) val_gril[j]-=v;
				 // 所有访问过的男生增加期望值
				if(book_boy[j]) val_boy[j]+=v;
				else slack[j]-=v;
			}
		}
	}
	 // 匹配完成 求出所有配对的好感度的和
	int ans=0;
	for(int i=0;i<n;i++) ans+=mmp[match[i]][i];
	return ans;
}

main函数

int main(){
	while(~scanf("%d",&n)){
		for(int i=0;i<n;i++)
			for(int j=0;j<n;j++){
				scanf("%d",&mmp[i][j]);
			}
			printf("%d\n",km());
	}
}

例题

1、HDU - 2255 奔小康赚大钱

AC代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=310;
const int inf=0x3f3f3f3f;
int mmp[maxn][maxn];
bool book_gril[maxn];
bool book_boy[maxn];
int val_gril[maxn];
int val_boy[maxn];
int match[maxn];
int slack[maxn];
int n;
bool dfs(int gril){
	book_gril[gril]=true;
	for(int boy=0;boy<n;boy++){
		if(book_boy[boy]) continue;
		int x=val_gril[gril]+val_boy[boy]-mmp[gril][boy];
		if(x==0){
			book_boy[boy]=true;
			if(match[boy]==-1||dfs(match[boy])){
				match[boy]=gril;
				return true;
			}
			
		}
		else{
			slack[boy]=min(slack[boy],x);
		}
	}
	return false;
}
int km(){
	memset(match,-1,sizeof(match));
	memset(val_boy,0,sizeof(val_boy));
	for(int i=0;i<n;i++){
		val_gril[i]=mmp[i][0];
		for(int j=1;j<n;j++){
			val_gril[i]=max(val_gril[i],mmp[i][j]);
		}
	}
	for(int i=0;i<n;i++){
		fill(slack,slack+n,inf);
		while(1){
			memset(book_gril,false,sizeof(book_gril));
			memset(book_boy,false,sizeof(book_boy));
			if(dfs(i)) break;
			int v=inf;
			for(int j=0;j<n;j++)if(!book_boy[j]) v=min(v,slack[j]);
			for(int j=0;j<n;j++){
				if(book_gril[j]) val_gril[j]-=v;
				if(book_boy[j]) val_boy[j]+=v;
				else slack[j]-=v;
			}
		}
	}
	int ans=0;
	for(int i=0;i<n;i++) ans+=mmp[match[i]][i];
	return ans;
}
int main(){
	while(~scanf("%d",&n)){
		for(int i=0;i<n;i++)
			for(int j=0;j<n;j++){
				scanf("%d",&mmp[i][j]);
			}
			printf("%d\n",km());
	}
}