km算法求带权二分图的最大匹配

做法

可以与匈牙利算法类比.
首先每一个匹配点都有一个期望匹配值为与它相连的所有边的最大搭配值,所有被匹配点的期望匹配值均为0.
然后如果匹配点与被匹配点的期望值之和恰好为该匹配值,则可以匹配,用匈牙利算法来求,如果没有找到则让参与此次失败匹配的点中的匹配点加上k,被匹配点减去k,k为产生一个合法匹配还需要的最小值(在代码中用need表示),全部匹配完后输出答案即可

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#define N 310
#define C ch=getchar()
#define INF 0x3f3f3f3f
using namespace std;

int n,m,mm[N][N],ex[2][N],mat[N],need[N],ans;
bool use[2][N];
char ch;

bool find(int now)
{
    register int i;
    use[0][now]=1;
    for(i=1;i<=n;i++)
    {
        if(use[1][i]) continue;
        int t=ex[0][now]+ex[1][i]-mm[now][i];
        if(t)
        {
            need[i]=min(need[i],t);
            continue;
        }
        use[1][i]=1;
        if(!mat[i]||find(mat[i]))
        {
            mat[i]=now;
            return 1;
        }
    }
    return 0;
}

inline int read()
{
    static int res;
    for(C;ch<'0';C);
    for(res=ch-48,C;ch>='0';res=res*10+ch-48,C);
    return res;
}

int main()
{
    int j;
    register int i,p;
    while(~scanf("%d",&n))
    {
        memset(ex[1],0,sizeof(ex[1]));
        memset(mat,0,sizeof(mat));
        for(i=1;i<=n;++i)
        {
            p=0;
            for(j=1;j<=n;++j)
            {
                mm[i][j]=read();
                p=max(p,mm[i][j]);
            }
            ex[0][i]=p;
        }
        for(i=1;i<=n;++i)
        {
            fill(need+1,need+n+1,INF);
            for(;;)
            {
                memset(use,0,sizeof(use));
                if(find(i)) break;
                p=INF;
                for(j=1;j<=n;++j) if(!use[1][j]) p=min(p,need[j]);
                for(j=1;j<=n;++j)
                {
                    if(use[0][j]) ex[0][j]-=p;
                    use[1][j]?ex[1][j]+=p:need[j]-=p;
                }
            }
        }
        ans=0;
        for(i=1;i<=n;++i)
        {
            ans+=mm[mat[i]][i];
        }
        printf("%d\n",ans);
    }
}