Uva11383 二分圖的完美匹配(深入理解KM演算法)

life4711發表於2015-10-09

https://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=2378

/**
Uva11383  二分圖的完美匹配(深入理解KM演算法)
劉汝佳大白書P351 少林決勝
*/
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <iostream>
using namespace std;
const int maxn=510;
const int inf=0x3f3f3f3f;

int n,nx,ny;
int link[maxn],lx[maxn],ly[maxn],slack[maxn];
int visx[maxn],visy[maxn],w[maxn][maxn];
int DFS(int x)
{
    visx[x]=1;
    for(int y=1;y<=ny;y++)
    {
        if(visy[y])continue;
        int t=lx[x]+ly[y]-w[x][y];
        if(t==0)
        {
            visy[y]=1;
            if(link[y]==-1||DFS(link[y]))
            {
                link[y]=x;
                return 1;
            }
        }
        else if(slack[y]>t)
        {
            slack[y]=t;
        }
    }
    return 0;
}

int KM()
{
    memset(link,-1,sizeof(link));
    memset(ly,0,sizeof(ly));
    for(int i=1;i<=nx;i++)
    {
        lx[i]=-inf;
        for(int j=1;j<=ny;j++)
        {
            if(w[i][j]>lx[i])
                lx[i]=w[i][j];
        }
    }
    for(int x=1;x<=nx;x++)
    {
        for(int i=1;i<=ny;i++)
            slack[i]=inf;
        while(1)
        {
            memset(visx,0,sizeof(visx));
            memset(visy,0,sizeof(visy));
            if(DFS(x))break;
            int d=inf;
            for(int i=1;i<=ny;i++)
            {
                if(!visy[i]&&d>slack[i])
                    d=slack[i];
            }
            for(int i=1;i<=nx;i++)
            {
                if(visx[i])
                    lx[i]-=d;
            }
            for(int i=1;i<=ny;i++)
            {
                if(visy[i])
                    ly[i]+=d;
                else
                    slack[i]-=d;
            }
        }
    }
    int res=0;
    for(int i=1;i<=ny;i++)
    {
        if(link[i]>-1)
            res+=w[link[i]][i];
    }
    return res;
}
int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                scanf("%d",&w[i][j]);
            }
        }
        nx=n,ny=n;
        int ans=KM();
       // printf("KM=%d\n",ans);
        int sum=0;
        for(int i=1;i<=n;i++)
        {
            printf("%d%c",lx[i]," \n"[i==n]);
            sum+=lx[i];
        }
        for(int i=1;i<=n;i++)
        {
            printf("%d%c",ly[i]," \n"[i==n]);
            sum+=ly[i];
        }
        printf("%d\n",sum);
    }
    return 0;
}


相關文章