poj 2195 KM算法

n个人要回到n个屋子，要求最小的花费。

KM算法可以求二分图最大权匹配，求最少花费可以先对距离取负数，求最大，再取负数即可。

ps：km算法看的还不是很懂啊《《《》》

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

#define MAXN 110
#define inf 999999999

struct point
{
    int x,y;
}pos_k[MAXN],pos_h[MAXN];
int n,m;
int w[MAXN][MAXN];
int lx[MAXN],ly[MAXN];
int linky[MAXN];
int visx[MAXN],visy[MAXN];
int slack[MAXN];
int nx,ny;

bool find(int x)
{
    visx[x] = true;
    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] = true;
            if(linky[y]==-1 || find(linky[y]))
            {
                linky[y] = x;
                return true;        //找到增广轨
            }
        }
        else if(slack[y] > t)
            slack[y] = t;
    }
    return false;                   //没有找到增广轨（说明顶点x没有对应的匹配，与完备匹配(相等子图的完备匹配)不符）
}

int KM()                //返回最优匹配的值
{
    int i,j;
    memset(linky,-1,sizeof(linky));
    memset(ly,0,sizeof(ly));
    for(i = 1; i <=nx; i++)
    {
        lx[i] = -inf;
         for(j = 1; j <=nx; j++)
            if(w[i][j] > lx[i])
                lx[i] = w[i][j];
    }

    for(int x = 1; x <=nx; x++)
    {
        for(i = 1; i <=nx; i++)
            slack[i] = inf;
        while(true)
        {
            memset(visx,0,sizeof(visx));
            memset(visy,0,sizeof(visy));
            if(find(x))                     //找到增广轨，退出
                break;
            int d = inf;
            for(i = 1; i <=nx; i++)          //没找到，对l做调整(这会增加相等子图的边)，重新找
            {
                if(!visy[i] && d > slack[i])
                    d = slack[i];
            }
            for(i = 1; i <=nx; i++)
            {
                if(visx[i])
                    lx[i] -= d;
            }
            for(i = 1; i <=ny; i++)
            {
                if(visy[i])
                     ly[i] += d;
                else
                    slack[i] -= d;
            }
        }
    }
    int result = 0;
    for(i = 1; i <=ny; i++)
    if(linky[i]>-1)
        result += w[linky[i]][i];
    return result;
}

int dist(point a,point b)
{
    int dis=0;
    dis+=(a.x-b.x)>0?(a.x-b.x):(b.x-a.x);
    dis+=(a.y-b.y)>0?(a.y-b.y):(b.y-a.y);
    return dis;
}

bool read()
{
    scanf("%d%d",&n,&m);
    getchar();
    char buf[MAXN];
    if(n+m==0)
        return false;
    nx=0,ny=0;
    for(int i=1;i<=n;i++)
    {
        gets(buf);
        for(int j=0;j<m;j++)
        {
            if(buf[j]=='H')
            {
                pos_h[ny].x=i;
                pos_h[ny++].y=j+1;
            }
            else if(buf[j]=='m')
            {
                pos_k[nx].x=i;
                pos_k[nx++].y=j+1;
            }
        }
    }
    for(int i=0;i<nx;i++)
        for(int j=0;j<ny;j++)
        {
            w[i+1][j+1]=-dist(pos_k[i],pos_h[j]);//取负
        }
    return true;
}

int main()
{
    while(read())
    {
        printf("%d\n",-KM());
    }
    return 0;
}