hdu 1533 Going Home KM算法

题目大意：

给你一张地图，上面有n个人，n座房子。一个人走到一座房子的

步数为他的花费。先让每个人走到不同的房子中，问是否有一种方案使得

总花费最小。KM算法经典题目，先算出每个人走到n座房子的花费作为

点到点的权值，然后权值取反，求完美匹配下的最大权值和，最后再将

返回的结果取反就是答案。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<vector>
#include<cmath>
#include<string>
#include<algorithm>
#include<set>
#include<map>
#include<cstring>
#include<queue>
#include<stack>
#include<list>
#define mp  make_pair


using namespace std;
typedef long long ll;
//const int MAXN=1000;
const ll mod=1e9+7;
int G[102][102];
int dis[102][102];
int flag[102][102];
int n,m;
int dir[4][2]={{1,0},{-1,0},{0,1},{0,-1}};

const int maxn = 310;
const int INF = 0x3f3f3f3f;
int nx,ny;//两边的点数
int g[maxn][maxn];//二分图描述，g[i][j]代表点i到点j边权，
//注意两边的点标号都是从0开始，与匈牙利算法网络流不同
int linker[maxn],lx[maxn],ly[maxn];//y中各点匹配状态，x,y中的点标号
int slack[maxn];
bool visx[maxn],visy[maxn];

bool DFS(int x)
{
    visx[x] = true;
    for(int y = 0; y < ny; y++)
    {
        if(visy[y])continue;
        int tmp = lx[x] + ly[y] - g[x][y];
        if(tmp == 0)
        {
            visy[y] = true;
            if(linker[y] == -1 || DFS(linker[y]))
            {
                linker[y] = x;
                return true;
            }
        }
        else if(slack[y] > tmp)
            slack[y] = tmp;
    }
    return false;
}
int KM()
{
    memset(linker,-1,sizeof(linker));
    memset(ly,0,sizeof(ly));
    for(int i = 0;i < nx;i++)
    {
        lx[i] = -INF;
        for(int j = 0;j < ny;j++)
            if(g[i][j] > lx[i])
                lx[i] = g[i][j];
    }
    for(int x = 0;x < nx;x++)
    {
        for(int i = 0;i < ny;i++)
            slack[i] = INF;
        while(true)
        {
            memset(visx,false,sizeof(visx));
            memset(visy,false,sizeof(visy));
            if(DFS(x))break;
            int d = INF;
            for(int i = 0;i < ny;i++)
                if(!visy[i] && d > slack[i])
                    d = slack[i];
            for(int i = 0;i < nx;i++)
                if(visx[i])
                    lx[i] -= d;
            for(int i = 0;i < ny;i++)
            {
                if(visy[i])ly[i] += d;
                else slack[i] -= d;
            }
        }
    }
    int res = 0;
    for(int i = 0;i < ny;i++)
        if(linker[i] != -1)
            res += g[linker[i]][i];
    return res;
}


void findmindis(int x,int y){
    memset(dis,-1,sizeof(dis));
    dis[x][y]=0;
    queue<pair<int,int> > que;
    que.push(mp (x,y));
    while(que.size()){
        pair<int,int > tmp=que.front();
        que.pop();
        int x1=tmp.first;
        int y1=tmp.second;
        for(int i=0;i<4;i++){
            int xx=x1+dir[i][0];
            int yy=y1+dir[i][1];
            if(xx>=0&&xx<n&&yy>=0&&yy<m&&dis[xx][yy]==-1){
                dis[xx][yy]=dis[x1][y1]+1;
                que.push(mp(xx,yy));
                if( G[xx][yy]==2 ){
                    g[ flag[x][y] ][ flag[xx][yy] ]=-dis[xx][yy];
                }
            }
        }
    }
}

int main()
{
    while(scanf("%d%d",&n,&m) !=EOF &&n!=0&&m!=0 )
    {
        nx=ny=0;
        memset(flag,-1,sizeof(flag));
        for(int i=0;i<n;i++){
            char s[102];
            scanf("%s",s);
            for(int j=0;j<m;j++){
                G[i][j]=0;
                if(s[j]=='m')
                {G[i][j]=1;flag[i][j]=nx++;}
                if(s[j]=='H')
                {G[i][j]=2;flag[i][j]=ny++;}
            }
        }
        for(int i=0;i<n;i++){
            fill( g[i],g[i]+m,-INF );
        }
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                if( G[i][j]==1 ){
                    findmindis(i,j);
                }
            }
        }
        printf("%d\n",-KM());
    }
    return 0;
}