POJ 2195 Going Home(二分图最优匹配)

POJ 2195 Going Home(二分图最优匹配)

http://poj.org/problem?id=2195

题意:

       给定一个N*M的地图，地图上有若干个man和house，且man与house的数量一致。man每移动一格需花费$1（即单位费用=单位距离），一间house只能入住一个man。现在要求所有的man都入住house，求最小费用。

分析:

       每个人对应二分图左边的点,每个房间对应二分图右边的点.每个人与每个房间都有一条带权值的边,表示这个人到该房间的花费.

       为了使得每个人都得到一个房间,且他们行走的开销最小.

       那么本题求得就是该二分图的一个完备匹配且该匹配边的权值和最小了.

       但是KM算法只能求二分图的最优匹配(即匹配边权值和最大的匹配),我们现在要求权值和最小的匹配怎么办?

       我们只需要把所有边的权值取负即可。

       假设存在一个最优(在所有解中花费ans最小，那么-ans自然最大)的解，那么当我们把所有权值取负之后，我们用KM算法得到的一个匹配的权值和必然就是那个最优的解(-ans最大)。因为KM算法不会漏掉权值和最大的那组解。(想想是不是)

       最终用KM算法求出的就是我们所需的最小权值匹配的负数.

AC代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
const int maxn=100+5;

//把W[maxn][maxn]数组的内容读进去之后，调用solve(n)即可计算出二分图最优匹配
//不过需要保证该图肯定有完美匹配
//因为本图用的W[][]来表示一个完全图，所以一定存在完美匹配的
struct Max_Match
{
    int W[maxn][maxn],n;  //W是权值矩阵,n为左右点集大小
    int Lx[maxn],Ly[maxn];//左右点集的可行顶标值
    bool S[maxn],T[maxn]; //标记左右点集是否已被访问过
    int left[maxn];       //left[i]=j表右i与左j匹配,为-1时表无匹配

    bool match(int i)
    {
        S[i]=true;
        for(int j=1;j<=n;j++)if(Lx[i]+Ly[j]==W[i][j] && !T[j])
        {
            T[j]=true;
            if(left[j]==-1 || match(left[j]))
            {
                left[j]=i;
                return true;
            }
        }
        return false;
    }

    //更新可行顶标,纳入更多的边进来
    void update()
    {
        int a=1<<30;
        for(int i=1;i<=n;i++)if(S[i])
        for(int j=1;j<=n;j++)if(!T[j])
        {
            a = min(a,Lx[i]+Ly[j]-W[i][j]);
        }
        for(int i=1;i<=n;i++)
        {
            if(S[i]) Lx[i]-=a;
            if(T[i]) Ly[i]+=a;
        }
    }

    int solve(int n)
    {
        this->n=n;
        memset(left,-1,sizeof(left));
        for(int i=1;i<=n;i++)//初始化可行顶标值
        {
            Lx[i]=Ly[i]=0;
            for(int j=1;j<=n;j++)
                Lx[i]=max(Lx[i], W[i][j]);
        }

        for(int i=1;i<=n;i++)
        {
            while(true)
            {
                for(int j=1;j<=n;j++) S[j]=T[j]=false;
                if(match(i)) break;
                else update();
            }
        }

        int ans=0;//最优完美匹配的权值
        for(int i=1;i<=n;i++) ans+= W[left[i]][i];
        return ans;
    }
}KM;
char s[maxn];
struct node
{
	int x,y;
	node(){}
	node(int a,int b):x(a),y(b){}
}house[maxn],man[maxn];
int main()
{
	int n,m;
	while(~scanf("%d%d",&n,&m)&&n&&m)
	{
		int cnth=0,cntm=0;
		for(int i=1;i<=n;i++)
		{
			scanf("%s",s);
			for(int j=0;j<m;j++)
			{
				if(s[j]=='H')
				{
					house[cnth++]=node(i,j+1);
				}
				else if(s[j]=='m')
				{
					man[cntm++]=node(i,j+1);
				}
			}
		}
		for(int i=0;i<cntm;i++)
			for(int j=0;j<cnth;j++)
			KM.W[i+1][j+1]=-(abs(house[j].x-man[i].x)+abs(house[j].y-man[i].y));
		printf("%d\n",-KM.solve(cnth));
	}
}