1.KM算法基础
每个员工做不同的工作效率不同,如何分配使总效率最大?
暴力:用匈牙利算法求出所有最大匹配,再从中找出效率和最高的
KM算法:解决带权二分图最优匹配
2.KM算法流程
1.为各顶点赋值值,将左顶点赋值为最大权值,右顶点赋值为0
2.用匈牙利算法寻找完备匹配
3.若未找到完备匹配则修改顶点权值
4.重复(2)(3)直到找到完备匹配为止
时间复杂度O(n^3)
3.KM算法演示
(1)
初始化各顶点值
将左顶点赋值为最大权
将右顶点赋值为0
——————————————————————————————————————————————
(2)
从A出发
尝试与a匹配,4+0!=3 失配
尝试与c匹配,4+0=4 匹配成功
——————————————————————————————————————————————
(3)
从B出发
尝试与a匹配,3+0!=2 失配
尝试与b匹配,3+0!=1 失配
尝试与c匹配,3+0=3 成功
但c已经与A匹配过
尝试修改A的匹配边,发现无法修改
此时B无法匹配
涉及的顶点为A,B,c
令所有左边点-1,所有右边点+1
——————————————————————————————————————————————
(4)
继续从B开始匹配
尝试与a匹配 2+0=2 成功
——————————————————————————————————————————————
(5)
从C开始匹配
5+1!=5 匹配失败
此时C无边可以匹配
涉及到的点为C
令所有左顶点-1,右顶点+1
修改完成后,继续从C开始匹配
尝试与c匹配 4+1=5 成功
但c已经与A匹配
尝试修改A的匹配,发现A-a可以匹配
但a已经和B匹配
尝试修改B的匹配,发现无法修改
此时C无法匹配
涉及的顶点为A,B,C,c,a
所有左边顶点值-1,右边顶点值+1
——————————————————————————————————————————————
(6)
继续从C开始匹配
尝试与c匹配 3+2=5 成功
但c已经与A匹配
尝试修改A的匹配,发现A-a可以匹配
但a已经与B匹配
尝试修改B的匹配,发现B-b可以匹配
将B的匹配修改为B-b
将A的匹配修改为A-a
此时,C与c成功匹配
——————————————————————————————————————————————
(7)
所有顶点都已被匹配
算法结束
此时最优匹配权值为
3+1+5=9
——————————————————————————————————————————————
4.KM算法实现
#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;
}