#include <iostream>
#include <cstring>
#include<math.h>
#include <cstdio>
#include <random>
using namespace std;
const int MAXN = 10000;
const double INF =pow(10,20);
int lenn;
int lenm;
double E_val[MAXN][MAXN]; // 记录每条边的权值
double L_val[MAXN]; // 每个左顶点的值
double R_val[MAXN]; // 每个右顶点的值
bool L_vis[MAXN]; // 记录每一轮匹配匹配过的左顶点
bool R_vis[MAXN]; // 记录每一轮匹配匹配过的右顶点
int match[MAXN]; // 记录每个右顶点匹配到的左顶点 如果没有则为-1
double slack[MAXN]; // 记录每个右顶点如果能被左顶点匹配最少还需要多少值
bool dfs(int now)
{
L_vis[now] = true;
for (int i = 0; i < lenm; i++)
{
if (R_vis[i]) continue; // 每一轮匹配 每个右顶点只尝试一次
double 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;
}
double KM()
{
memset(match, -1, sizeof(match)); // 初始每个右顶点都没有匹配的左顶点
memset(R_val, 0, sizeof(R_val)); // 初始每个右顶点的值为0
for (int i = 0; i < lenm; ++i) { //初始每个左顶点为与其相连边的最大权值
L_val[i] = E_val[i][0];
for (int j = 1; j < lenm; ++j) {
L_val[i] = max(L_val[i], E_val[i][j]);
}
}
// 尝试为每一个左顶点匹配
for (int i = 0; i < lenm; ++i) {
fill(slack, slack + lenm, INF); // 因为要取最小值 初始化为无穷大
while (1) {
// 为每个左顶点匹配的方法是 :如果找不到就降低期望值,直到找到为止
// 记录每轮匹配中左右顶点是否被尝试匹配过
memset(L_vis, false, sizeof(L_vis));
memset(R_vis, false, sizeof(R_vis));
if (dfs(i)) break; // 找到匹配 退出
// 如果不能找到 就降低期望值
// 最小可降低的期望值
double d = INF;
for (int j = 0; j < lenm; ++j)
if (!R_vis[j]) d = min(d, slack[j]);
for (int j = 0; j < lenm; ++j) {
// 所有访问过的(被涉及的)左顶点降低值
if (L_vis[j]) L_val[j] -= d;
// 所有访问过(被涉及的)的右顶点增加值
if (R_vis[j]) R_val[j] += d;
// 没有访问过的右顶点 因为左顶点的期望值降低,距离被左顶点匹配又进了一步
else slack[j] -= d;
}
}
}
// 匹配完成 求出所有匹配的权值和
double res = 0;
int count=0;
for (int i = 0; i < lenm; ++i)
if(E_val[ match[i] ][i]!=-INF){
count++;
res += E_val[ match[i] ][i];
}
cout<<count<<endl;
return res;
}
int main()
{
lenm=10;
lenn=5;
for (int i = 0; i < lenn; ++i){
for (int j = 0; j < lenm; ++j){
E_val[i][j]=-1.0*rand()/RAND_MAX *(1.0);
}
}
for (int i = lenn; i < lenm; ++i){
for (int j = 0; j < lenm; ++j){
E_val[i][j]=-INF;
}
}
for(int i=0;i<lenm;i++){
for(int j=0;j<lenm;j++){
cout<<E_val[i][j];
}
cout<<endl;
}
cout<< -KM()<<endl;
return 0;
}
KM算法(Kuhn-Munkres),用来求带实数权二分图的最大权匹配(最优匹配),且左右两边的顶点数不同
最新推荐文章于 2024-07-25 21:53:52 发布