洛谷---P3967 [TJOI2014]匹配---最大二分带权匹配---KM算法

题目描述

有 $N$ 个单身的男孩和 $N$ 个单身女孩，男孩 $i$ 和女孩 $j$ 在一起得到的幸福值为 Hi,jH_{i,j}Hi,j​。

一个匹配即对这 $N$ 个男孩女孩的安排：每个男孩恰好有一个女朋友，每个女孩恰好有一个男朋友。

一个匹配的幸福值即这 $N$ 对男女朋友的幸福值的和。

经典的问题是计算幸福值最大的匹配，即完美匹配。然而完美匹配有时候并不唯一，你需要计算对于所有的完美匹配，其交集是什么。

输入格式

输入的第一行是一个正整数 $N$ 。

接下来是一个 $N\times N$ 大小的矩阵 $H$，Hi,jH_{i,j}Hi,j​ 表示男孩 $i$ 和女孩 $j$ 在一起的幸福值。

输出格式

第一行输出完美匹配的幸福值，接下来是若干行，每一行是一对整数 $i$ 和 $j$，表示男孩 $i$ 和女孩 $j$ 在所有完美匹配的交集中，以 $i$ 的递增顺序输出。

输入输出样例

输入 #1 复制

3
1 1 1
2 1 1
1 1 1

输出 #1 复制

4
2 1

说明/提示

• 对于 $30\%$ 的数据，$N\le 30$；
• 对于 $100\%$ 的数据，$1\le N\le 80$，0≤Hi,j≤5×1030\leq H_{i,j}\leq 5\times10^30≤Hi,j​≤5×103。

实现代码

#include<iostream>
#include<cstring>
#include<algorithm>

using namespace std;
const int maxn = 105;
const int inf = 0x3f3f3f3f;

int	link[maxn][maxn];	
int ex_boy[maxn], ex_girl[maxn];	
bool vis_boy[maxn], vis_girl[maxn];		
int match[2][maxn];		
int slack[maxn];		
int n;

typedef pair<int, int> Mate;
Mate mate[maxn];
int cnt = 0;
bool cmp(Mate m1, Mate m2) { return m1.first < m2.first; }


bool dfs(int boy, int cnt) {
	vis_boy[boy] = true;
	for (int girl = 1; girl <= n; girl++) {
		if (vis_girl[girl]) continue;
		int dis = ex_boy[boy] + ex_girl[girl] - link[boy][girl];
		if (dis == 0) {
			vis_girl[girl] = true;
			if (!match[cnt][girl] || dfs(match[cnt][girl], cnt)) {
				match[cnt][girl] = boy;
				return true; // 匹配成功
			}
		}
		else slack[girl] = min(slack[girl], dis);
	}
	return false;	// 匹配失败返回失败
}

int km(int cnt) {
	for (int i = 1; i <= n; i++) {
		match[cnt][i] = ex_girl[i] = 0;
		ex_boy[i] = link[i][1];
		for (int j = 2; j <= n; j++) ex_boy[i] = max(ex_boy[i], link[i][j]); 
	}
	for (int boy = 1; boy <= n; boy++) {
		for (int i = 1; i <= n; i++) slack[i] = inf;
		while (true) {
			memset(vis_boy, false, sizeof(vis_boy)); 
			memset(vis_girl, false, sizeof(vis_girl));
				
			if (dfs(boy, cnt)) break; // 如果匹配成功就可以爬了
			// 没匹配成功就去降低期望值，因为初始化是最大期望，所以只可以降

			int d = inf; // 最小减低的期望值
			for (int i = 1; i <= n; i++) if(!vis_girl[i]) d = min(slack[i], d);

			for (int i = 1; i <= n; i++) {
				// 修改匹配过的两者之间的期望值
				if (vis_boy[i]) ex_boy[i] -= d; 
				if (vis_girl[i]) ex_girl[i] += d;
				else slack[i] -= d;		//没有匹配的最小期望差值要下降
			}
		}
	}
	int ans = 0;
	for (int i = 1; i <= n; i++) ans += link[match[cnt][i]][i]; 
	return ans;
}

int main() { 
	ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
	while (cin >> n) {
		int max_ans = 0;
		for (int i = 1; i <= n; i++) for (int j = 1; j <= n; j++) cin >> link[i][j];
		cout << (max_ans = km(0)) << endl;

		for (int i = 1; i <= n; i++) {
			int tmp = link[match[0][i]][i]; // 记录第一次的匹配期望
			link[match[0][i]][i] = 0;
			if (km(1) != max_ans) mate[cnt++] = Mate(match[0][i], i);
			link[match[0][i]][i] = tmp;
		}

		sort(mate, mate + cnt, cmp);
		for (int i = 0; i < cnt; i++) cout << mate[i].first << " " << mate[i].second << endl;
	}
	return 0;
}