KM算法(Kuhn-Munkres),用来求带实数权二分图的最大权匹配(最优匹配),且左右两边的顶点数不同

#include <iostream>
#include <cstring>
#include<math.h>
#include <cstdio>
#include <random>
using namespace std;
const int MAXN = 10000;
const double eps = 1e-6;
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 (fabs(tmp) < eps) {  // 如果符合要求
            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]!=0){
        count++;
        res += 1.1-E_val[ match[i] ][i];
        cout<<match[i]<<' '<<i<<endl;
    }
    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);
               cout<<E_val[i][j]<<' ';
            }
            cout<<endl;
            
        }
        for (int i = 0; i < lenn; ++i){
            for (int j = 0; j < lenm; ++j){
               E_val[i][j]=1.1-E_val[i][j];
            }
            
        }
        for (int i = lenn; i < lenm; ++i){
            for (int j = 0; j < lenm; ++j){
               E_val[i][j]=0;
            }
            
        }

        
 
        cout<< KM()<<endl;
    return 0;
}
 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值