KM算法萌新讲解篇

KM算法

 

首先了解问题：也就是最大权值匹配；

二分图里，边带了权值，求整幅图里匹配最大/最小的权值

因为接触匈牙利算法的时候看的是找对象系列的博文，所以也自己写一发找对象的博文吧；

算法背景：

信息学院计算机某班级有５位玉树凌风的男子：小诸，小包，小许，小应，小章；

外语学院英语系某班级有５位国色天香的女子：小四，小雨，小美，小丽，小英；

名称不重要！强行变成Ｘ，Ｙ，２３３３３３

然后５位男子对５位女子都有一个好感度，见表；

	ｙ０	ｙ１	ｙ２	ｙ３	ｙ４
ｘ０	４	１	６	２	３
ｘ１	５	０	３	７	６
ｘ２	２	３	４	５	８
ｘ３	３	４	６	３	４
ｘ４	４	６	５	８	６

现在要给这几个男孩子找女朋友～使得他们所有好感度和最大。

KM算法步骤：

初始情况下把好感度都放在男子的集合上，即

ＬＸ０＝６;

ＬＸ１＝７;

ＬＸ２＝８;

ＬＸ３＝６;

ＬＸ４＝８;

然后枚举Ｘ集合元素去匹配，如果匹配到的边ｗｉｊ＝ＬＸｉ＋ＬＹｊ，并且ｊ还不存在配偶，那么，嘿嘿嘿，就让他们牵手～(✿◡‿◡)

比如：在枚举Ｘ集合时，Ｘ０－Ｙ２；Ｘ１－Ｙ３；Ｘ２－Ｙ４；

但是，有些女孩子很受欢迎，突然到了Ｘ３，Ｘ３最喜欢Ｙ２，而Ｙ２已经有Ｘ０喜欢了，窝们的目的是让整个好感度最大；

那么就会有两种情况：

①  ：我让Ｘ０再去找一个，然后让Ｘ３－Ｙ２；

②  ：我让Ｘ３另外去选，然后还是让Ｘ０－Ｙ２；

那么窝们怎么能随随便便让男孩子另外去找一个呢，窝们其实很容易想到一个方案就是让一个男孩子去找另外一个女孩子使得好感度比原先大的好感度降低的越少，那么这样还是会让总好感度最大；

具体的方法就是窝们可以开一个差值数组代表最大的好感度和其他好感度的差；

然后每次在寻找女朋友的时候将这个差值数组进行更新；

每次扫完女孩子以后，如果没有找到心仪对象，

找到那个最小差值ｄ

把ＬＸ数组种那些有矛盾的男孩子的值－＝ｄ；

把ＬＹ数组种那些引起矛盾的女孩的值＋＝ｄ；

观察窝们说一个男孩子找到女朋友的条件是：如果匹配到的边ｗｉｊ＝ＬＸｉ＋ＬＹｊ，并且ｊ还不存在配偶，那么，嘿嘿嘿，就让他们牵手～(✿◡‿◡)

这样就能保证在下次找配偶的过程中

１.LX-=d，当再次对当前有矛盾的男孩子寻找时可能一下子找到那个好感度减少最少的女孩，因为ＬＸ减少，然而ＬＹ不变，所以一旦满足条件，就返回ｔｒｕｅ；

２.LY+=d，引起矛盾的那个女生还是照样引起矛盾或者在之前就被寻找过在下次寻找到时不寻找；

３.在对出现矛盾的那个女生的本来配好的男朋友进行寻找时，同理１的情况，找到那个好感度减少最少的女孩且满足情况就返回；

这样算法流程说明（个人理解）：

ＬＸ的作用是维护权值减少最少，ＬＹ的作用是维护权值最大的依旧最大；

教科书术语：

（１）  令（ｕ，ｖ）是一个覆盖，如对于每个ｊ，uｉ＝ｍａｘ（ｗｉｊ），ｖｉ＝０；

（２）  在Ｋｕ＊ｖ中找出一个最大匹配Ｍ。如果Ｍ是一个完美匹配，停止并将Ｍ作为最大权值匹配返回。否则，令Ｑ是Ｋｕ＊ｖ中大小为Ｍ的一个定点覆盖。设Ｒ＝Ｑ∩Ｘ，Ｔ＝Ｑ∩Ｙ。令ｅ＝ｍｉｎ｛ｕｉ＋ｖｊ－ｗｉｊ，ｘｉ∈Ｘ－Ｒ，ｙｉ∈Ｙ－Ｔ｝。

（３）  对于ｘｉ∈Ｘ－Ｒ，从ｕｉ中减去ｅ．对ｙｉ∈Ｔ，把ｅ加到ｖｊ上，形成一个新的相等子图，转向步骤（２）；

贴一发模板：

bool Findpath(int u)
{
    vx[u]=1;
    for(int i=1;i<=n;i++)
    {
        if(vy[i]) continue;
        int temp=lx[u]+ly[i]-ma[u][i];
        if(temp==0)
        {
            vy[i]=1;
            if(match[i]==-1||Findpath(match[i]))
            {
                match[i]=u;
                return true;
            }
        }
        else if(d>temp)
            d=temp;
    }
    return false;
}

int KM()
{
    memset(match,-1,sizeof(match));
    memset(lx,0,sizeof(lx));
    memset(ly,0,sizeof(ly));

    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++)
        lx[i]=max(ma[i][j],lx[i]);

    for(int i=1;i<=n;i++)
    {
        while(1)
        {
            memset(vx,0,sizeof(vx));
            memset(vy,0,sizeof(vy));
            d=0x3f3f3f3f;
            if(Findpath(i)) break;
            for(int j=1;j<=n;j++)
            {
                if(vx[j]) lx[j]-=d;
                if(vy[j]) ly[j]+=d;
            }
        }
    }
    int res=0;
    for(int i=1;i<=n;i++)
    {
        res+=ma[match[i]][i];
    }
    return res;
}