KM算法详解+模板

原文地址: http://www.cnblogs.com/wenruo/p/5264235.html

KM算法用来求二分图最大权完美匹配。

 

本文没有给出KM算法的原理,只是模拟了一遍算法的过程。另,博主水平较差,发现问题欢迎指出,谢谢!!!!

 

现在有N男N女,有些男生和女生之间互相有好感,我们将其好感程度定义为好感度,我们希望把他们两两配对,并且最后希望好感度和最大。

KM算法详解-wenr

怎么选择最优的配对方法呢?

首先,每个女生会有一个期望值,就是与她有好感度的男生中最大的好感度。男生呢,期望值为0,就是……只要有一个妹子就可以啦,不挑~~

这样,我们把每个人的期望值标出来。

KM算法详解-wenr

接下来,开始配对。

配对方法:

我们从第一个女生开始,分别为每一个女生找对象。

每次都从第一个男生开始,选择一个男生,使男女两人的期望和要等于两人之间的好感度

注意:每一轮匹配每个男生只会被尝试匹配一次!

 

具体匹配过程:

==============为女1找对象===============

(此时无人配对成功)

根据 “男女两人的期望和要等于两人之间的好感度”的规则

女1-男1:4+0 != 3

女1-男3:4+0 == 4

所以女1选择了男3

女1找对象成功

==============为女1找对象成功============

 

==============为女2找对象===============

(此时女1—男3)

根据配对原则,女2选择男3

男3有主女1,女1尝试换人

我们尝试让女1去找别人

尝试失败

为女2找对象失败!

==============为女2找对象失败============

 

这一轮参与匹配的人有:女1,女2,男3。

怎么办???很容易想到的,这两个女生只能降低一下期望值了,降低多少呢?

任意一个参与匹配女生能换到任意一个这轮没有被选择过的男生所需要降低的最小值

比如:女1选择男1,期望值要降低1。 女2选择男1,期望值要降低1。 女2选择男2,期望值要降低2。

于是,只要期望值降低1,就有妹子可能选择其他人。所以妹子们的期望值要降低1点。

同时,刚才被抢的男生此时非常得意,因为有妹子来抢他,于是他的期望值提高了1点(就是同妹子们降低的期望值相同)。

于是期望值变成这样(当然,不参与刚才匹配过程的人期望值不变)

KM详解-wenr

==============继续为女2找对象=============

(此时女1—男3)

女2选择了男1

男1还没有被配对

女2找对象成功!

==============为女2找对象成功=============

 

==============为女3找对象===============

(此时女1—男3,女2-男1)

女3没有可以配对的男生……

女3找对象失败

==============为女3找对象失败============

此轮只有女3参与匹配

此时应该为女3降低期望值

降低期望值1的时候,女3-男3可以配对,所以女3降低期望值1

KM算法详解

 

==============继续为女3找对象============

(此时女1—男3, 女2-男1)

女3相中了男3

此时男3已经有主女1,于是女1尝试换人

女1选择男1

而男1也已经有主女2,女2尝试换人

前面说过,每一轮匹配每个男生只被匹配一次

所以女2换人失败

女3找对象再次失败

==============为女3找对象失败============

这一轮匹配相关人员:女1,女2,女3,男1,男3

此时,只要女2降低1点期望值,就能换到男2

(前面提过 只要任意一个女生能换到任意一个没有被选择过的男生所需要降低的最小值)

我们把相应人员期望值改变一下

KM算法详解-wenr

 

==============还是为女3找对象============

(此时女1—男3, 女2-男1)

女3选择了男3

男3有主女1,女1尝试换人

女1换到了男1

男1已经有主女2,女2尝试换人

女2换人男2

男2无主,匹配成功!!!

==============为女3找对象成功=============

匹配成功!!!撒花~~

到此匹配全部结束

此时

女1-男1,女2-男2,女3-男3

好感度和为最大:9

 

 

虽然不停换人的过程听起来很麻烦,但其实整个是个递归的过程,实现起来比较简单。比较复杂的部分就是期望值的改变,但是可以在递归匹配的过程中顺带求出来。

 

模板(带详细注释)(入门题:HDU2255(复杂度应该是O(N^3)

#include<bits/stdc++.h>
#define exp 1e-8
#define mian main
#define pii pair<int,int>
#define pll pair<ll,ll>
#define ll long long
#define pb push_back
#define PI  acos(-1.0)
#define inf 0x3f3f3f3f
#define w(x) while(x--)
#define int_max 2147483647
#define lowbit(x) (x)&(-x)
#define gcd(a,b) __gcd(a,b)
#define pq(x)  priority_queue<x>
#define ull unsigned long long
#define sc(x) scanf("%d",&x)
#define scl(x) scanf("%lld",&x)
#define pl(a,n) next_permutation(a,a+n)
#define ios ios::sync_with_stdio(false)
#define met(a,x) memset((a),(x),sizeof((a)))
using namespace std;
const int N=310;
int n;
int e[N][N]; //关系
int ex_girl[N];//女生的期望值
int ex_boy[N];//男生的期望值
int match[N]; //记录每个男生匹配到的女生,如果没有为-1
bool girl[N];//标记每一轮匹配过的女生
bool boy[N];//标记每一轮匹配过的男生
int slack[N];//记录每个男生要被女生选中还需多少期望值
bool line(int x)
{
    girl[x]=1;
    for(int i=1;i<=n;i++){
        if(boy[i])
            continue;
        int k=ex_boy[i]+ex_girl[x]-e[x][i];
        if(k==0){ //符合要求
            boy[i]=1;
            if(match[i]==-1||line(match[i])){
                match[i]=x;
                return true;
            }
        }
        else slack[i]=min(slack[i],k);
    }
    return false;
}
int km()
{
    met(ex_boy,0);
    met(match,-1);
    for(int i=1;i<=n;i++){
        ex_girl[i]=e[i][1];
        for(int j=2;j<=n;j++)
            ex_girl[i]=max(ex_girl[i],e[i][j]);  //初始每个女生的期望值为对每个男生的期望的最大值
    }
    for(int i=1;i<=n;i++){  //为每一个女生匹配
        met(slack,inf);
        while(1){
            met(girl,0);
            met(boy,0);
            if(line(i))  //匹配成功,退出
                break;
            int d=inf; //匹配失败
            for(int j=1;j<=n;j++)
                if(!boy[j])
                d=min(d,slack[j]);  //最小可降低的期望值
            for(int j=1;j<=n;j++){
                if(girl[j])
                    ex_girl[j]-=d;  //所以找到过的女生降低期望值
                if(boy[j])
                    ex_boy[j]+=d; //所有找到过的男生增加期望值
                else slack[j]-=d; //没有访问过的boy 因为girl们的期望值降低,距离得到女生倾心又进了一步!
            }
        }
    }
    int ans=0;
    for(int i=1;i<=n;i++)
        ans+=e[match[i]][i];
    return ans;
}
int main()
{
    while(~scanf("%d",&n)){
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
            scanf("%d",&e[i][j]);
        printf("%d\n",km());
    }
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章