题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1533
题解:要求的是最小权值之和,KM求的是最大权值之和。在建图的时候,将每条边的权值变为负数。然后lx[i]初始化为-INF,结果输出-ans,就可以得到最小权值。
#include <stdio.h>
#include <string.h>
#include <math.h>
#define INF 0x3f3f3f3f
#define MAXN 102
int w[MAXN][MAXN],match[MAXN];
int lx[MAXN],ly[MAXN],slack[MAXN];
int visitx[MAXN],visity[MAXN];
int nx,ny;
int find(int x)
{
int i,temp;
visitx[x]=1;
for(i=0;i<ny;++i)
{
if(visity[i])
continue;
temp=lx[x]+ly[i]-w[x][i];
if(temp==0)
{
visity[i]=1;
if(match[i]==-1||find(match[i]))
{
match[i]=x;
return 1;
}
}
else if(slack[i]>temp)
{
slack[i]=temp;
}
}
return 0;
}
int KM()
{
int i,j,d,sum;
memset(ly,0,sizeof(ly));
memset(match,-1,sizeof(match));
for(i=0;i<nx;++i)
{//lx初始化为与它关联边中最大的
lx[i]=w[i][0];
for(j=1;j<ny;++j)
if(w[i][j]>lx[i])
lx[i]=w[i][j];
}
for(i=0;i<nx;++i)
{
for(j=0;j<ny;++j)
slack[j]=INF;
while(1)
{
memset(visitx,0,sizeof(visitx));
memset(visity,0,sizeof(visity));
if(find(i))
break;
d=INF;
for(j=0;j<ny;++j)
{
if(!visity[j]&&d>slack[j])
d=slack[j];
}
for(j=0;j<nx;++j)
{
if(visitx[j])
lx[j]-=d;
}
for(j=0;j<ny;++j)
{
if(visity[j])
ly[j]+=d;
else
slack[j]-=d;
}
}
}
sum=0;
for(i=0;i<ny;++i)
{
if(match[i]!=-1)
sum+=w[match[i]][i];
}
return sum;
}
int main()
{
int i,j,mennum,housenum,m,n,ans;
char mp[MAXN];
int men[MAXN][2],house[MAXN][2];
while(scanf("%d%d",&n,&m)&&n)
{
mennum=housenum=0;
for(i=0;i<n;++i)
{
scanf("%s",mp);
for(j=0;mp[j]!='\0';++j)
{
if(mp[j]=='m')
{
men[mennum][0]=i;
men[mennum++][1]=j;
}
else if(mp[j]=='H')
{
house[housenum][0]=i;
house[housenum++][1]=j;
}
}
}
for(i=0;i<mennum;++i)
{
for(j=0;j<housenum;++j)
w[i][j]=-(abs(men[i][0]-house[j][0])+abs(men[i][1]-house[j][1]));
}
nx=mennum;
ny=housenum;
ans=KM();
printf("%d\n",~ans+1);//结果取反
}
return 0;
}