奔小康赚大钱 HDU - 2255 （KM算法模板）

原创于 2019-07-24 21:19:49 发布 · 241 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#KM

KM算法专栏收录该内容

5 篇文章

订阅专栏

本文详细介绍了KM算法的原理及其实现过程，并通过一个具体的村庄房屋分配案例来展示如何使用KM算法解决匹配问题以达到最大收益。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目连接

题目大意：

一个村内，n 家老百姓， n 间房子，然后每家老百姓对不同的房子出的价钱不同，然后问题就是村领导怎样分配才能使收入最大

思路：

裸的 KM 算法

AC代码

#include<iostream>
#include<cstring>
#include<stdio.h>
#include<math.h>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;
const int maxn = 505;
const int INF = 0x3f3f3f3f;
int G[maxn + 5][maxn + 5];
int ex_X[maxn], ex_Y[maxn];
bool vis_X[maxn], vis_Y[maxn];
int Slack[maxn + 5];
int Match[maxn + 5];
int N;
bool Dfs(int x){
	vis_X[x] = 1;
	for(int i = 1; i <= N; ++i){
		if(vis_Y[i]) continue;
		int tmp = ex_X[x] + ex_Y[i] - G[x][i];
		if(tmp != 0){
			Slack[i] = min (Slack[i], tmp);
		} else {
			vis_Y[i] = 1;
			if (Match[i] == -1 || Dfs(Match[i])) {
				Match[i] = x;
				return true;
			}
		}
	}
	return false;
}
/*
2 
100 10
15 23
*/

int KM(){
	memset(Match, -1, sizeof(Match));	
	memset(ex_X, 0, sizeof(ex_X));
	memset(ex_Y, 0, sizeof(ex_Y));
	
	for (int i = 1; i <= N; ++i) {
		for (int j = 1; j <= N; ++j) {
			ex_X[i] = max(ex_X[i], G[i][j]);
		}
	}
	for (int i = 1; i <= N; ++i) {
		memset(Slack, INF, sizeof Slack);
		while(1){
			memset(vis_X, 0, sizeof(vis_X));
			memset(vis_Y, 0, sizeof(vis_Y));
			if (Dfs(i)) break;
			int tmp = INF;
			for (int j = 1; j <= N; ++j) {
				if( vis_Y[j] == 0 )
					tmp = min(tmp, Slack[j]); 
			}
			if(tmp == INF) return -1;
			for(int j = 1; j <= N; ++j) {
				if(vis_X[j]) ex_X[j] -= tmp; 
				if(vis_Y[j]) ex_Y[j] += tmp;
				else Slack[j] -= tmp;
			}
		}
	}
	int Ans = 0;
	for(int i = 1; i <= N; i++){
		if(Match[i] != -1)
			Ans += G[Match[i]][i];
	}
	return Ans;
}
int main (){
	while(~scanf("%d", &N)){
		int Temp;
		for(int i = 1; i <= N; ++i){
			for(int j = 1; j <= N; ++j){
				scanf("%d", &G[i][j]);
			}
		}
		printf("%d\n", KM());
	}
	return 0;
}