KM算法学习小记

问题描述

给定一个二分图，其中每条边都有一个权值，求图的最大权完备匹配。
完备匹配指ans=|A|=|B|，其中ans为二分图的最大匹配，A、B是二分图两个互不相交的点集。

这个问题可以直接跑费用流，当然还有另一种方法，就是KM算法。

主要实现

为每个点设一个顶标Li，当Li+Lj=Wi,j时才能选择边(i,j)，不能选择时Li+Lj>Wi,j。
然后像匈牙利一样，枚举每个A集合的点，然后去找增广轨。
如果当前做到点i，找不到增广轨了，那么令S为i找增广轨遍历的A集里的点的集合，T为遍历B集的点的集合。那么确定一个值dx=min{Li+Lj-Wi,j}（i∈S j∉T）
然后给Li减dx（i∈S），Lj加dx（j∈T）（这样保证了原本可以匹配的边还能匹配，不能匹配的边中最大权的可以匹配）
最后所有L的和就是答案了。
初始时Li=max{wi,j}（i∈A）

其它问题

如果一开始二分图给出不是完备匹配呢？
把没有的边加上，令它们的权值为0即可

时间复杂度

找一次增广轨最坏情况复杂度为O(m)，L数组最多调整m次，所以总复杂度为$O(m^2)$。事实上一般没有这么大。

主要部分：

bool dfs(int x)
{
    if (vx[x]) return 0;
    vx[x]=1;
    for (int i=0;i<N;i++) if (!vy[i])
    {
        int t=L[0][x]+L[1][i]-val[x][i];
        if (!t)
        {
            vy[i]=1;
            if (f[i]<0 || dfs(f[i]))
            {
                f[i]=x;
                return 1;
            }
        }else s[i]=min(s[i],t);
    }
    return 0;
}

int km()
{
    memset(L[1],0,sizeof(L[1]));
    for (int i=0;i<N;i++) L[0][i]=-1e9;
    for (int i=0;i<N;i++)
    {
        for (int j=0;j<N;j++)
        {
            L[0][i]=max(L[0][i],val[i][j]);
        }
    }
    memset(f,255,sizeof(f));
    for (int i=0;i<N;i++)
    {
        memset(s,127,sizeof(s));
        for (memset(vx,0,sizeof(vx)),memset(vy,0,sizeof(vy));!dfs(i);memset(vx,0,sizeof(vx)),memset(vy,0,sizeof(vy)))
        {
            int dx=1e9;
            for (int j=0;j<N;j++) if (!vy[j]) dx=min(dx,s[j]);
            for (int j=0;j<N;j++)
            {
                if (vx[j]) L[0][j]-=dx;
                if (vy[j]) L[1][j]+=dx;else s[j]-=dx;//
            }
        }
    }
    int ans=0;
    for (int i=0;i<N;i++) ans+=val[f[i]][i];
    return ans;
}