#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
const int MAXN = 305;
const int INF = 0x3f3f3f3f;
int E_val[MAXN][MAXN]; // 记录每条边的权值
int L_val[MAXN]; // 每个左顶点的值
int R_val[MAXN]; // 每个右顶点的值
bool L_vis[MAXN]; // 记录每一轮匹配匹配过的左顶点
bool R_vis[MAXN]; // 记录每一轮匹配匹配过的右顶点
int match[MAXN]; // 记录每个右顶点匹配到的左顶点 如果没有则为-1
int slack[MAXN]; // 记录每个右顶点如果能被左顶点匹配最少还需要多少值
int N; //顶点数
bool dfs(int now)
{
L_vis[now] = true;
for (int i = 0; i < N; i++)
{
if (R_vis[i]) continue; // 每一轮匹配 每个右顶点只尝试一次
int tmp = L_val[now] + R_val[i] - E_val[now][i];
if (tmp == 0) { // 如果符合要求
R_vis[i] = true;
if (match[i] == -1 || dfs( match[i] )) { // 找到一个没有匹配的右顶点 或者该右顶点当前匹配的左顶点可以找到其它匹配
match[i] = now;
return true;
}
} else {
slack[i] = min(slack[i], tmp); // slack 可以理解为该右顶点能被一个左顶点匹配 还需多少值 取最小值
}
}
return false;
}
int KM()
{
memset(match, -1, sizeof(match)); // 初始每个右顶点都没有匹配的左顶点
memset(R_val, 0, sizeof(R_val)); // 初始每个右顶点的值为0
for (int i = 0; i < N; ++i) { //初始每个左顶点为与其相连边的最大权值
L_val[i] = E_val[i][0];
for (int j = 1; j < N; ++j) {
L_val[i] = max(L_val[i], E_val[i][j]);
}
}
// 尝试为每一个左顶点匹配
for (int i = 0; i < N; ++i) {
fill(slack, slack + N, INF); // 因为要取最小值 初始化为无穷大
while (1) {
// 为每个左顶点匹配的方法是 :如果找不到就降低期望值,直到找到为止
// 记录每轮匹配中左右顶点是否被尝试匹配过
memset(L_vis, false, sizeof(L_vis));
memset(R_vis, false, sizeof(R_vis));
if (dfs(i)) break; // 找到匹配 退出
// 如果不能找到 就降低期望值
// 最小可降低的期望值
int d = INF;
for (int j = 0; j < N; ++j)
if (!R_vis[j]) d = min(d, slack[j]);
for (int j = 0; j < N; ++j) {
// 所有访问过的(被涉及的)左顶点降低值
if (L_vis[j]) L_val[j] -= d;
// 所有访问过(被涉及的)的右顶点增加值
if (R_vis[j]) R_val[j] += d;
// 没有访问过的右顶点 因为左顶点的期望值降低,距离被左顶点匹配又进了一步
else slack[j] -= d;
}
}
}
// 匹配完成 求出所有匹配的权值和
int res = 0;
for (int i = 0; i < N; ++i)
res += E_val[ match[i] ][i];
return res;
}
int main()
{
while (~scanf("%d", &N)) {
for (int i = 0; i < N; ++i)
for (int j = 0; j < N; ++j)
scanf("%d", &E_val[i][j]);
printf("%d\n", KM());
}
return 0;
}
km算法带权二分图匹配
最新推荐文章于 2022-12-10 15:20:48 发布