题目:http://acm.hdu.edu.cn/showproblem.php?pid=1533
求二分图的最优匹配。 求最小费用时,只要将所有费用全部变成负数,然后按照KM算法求最大值。对得到的结果进行取反就可以了。
下面是AC代码:
#include<cstdio>
#include<cstring>
#include<cstdlib>
using namespace std;
const int maxn = 105;
const int INF = (1<<30)-1;
int g[maxn][maxn];
int lx[maxn],ly[maxn];
int match[maxn];
bool visx[maxn],visy[maxn];
int slack[maxn];
char map[maxn][maxn];
int n,m;
int mn,hn;
struct node{
int x,y;
}man[maxn],house[maxn];
int dit(node a,node b){
return abs(a.x-b.x)+abs(a.y-b.y);
}
bool dfs(int cur){
visx[cur] = true;
for(int y=1;y<=hn;y++){
if(visy[y]) continue;
int t=lx[cur]+ly[y]-g[cur][y];
if(t==0){
visy[y] = true;
if(match[y]==-1||dfs(match[y])){
match[y] = cur;
return true;
}
}
else if(slack[y]>t){
slack[y]=t;
}
}
return false;
}
int KM(){
memset(match,-1,sizeof(match));
memset(ly,0,sizeof(ly));
for(int i=1 ;i<=mn;i++){
lx[i]=-INF;
for(int j=1;j<=hn;j++)
if(g[i][j]>lx[i]) lx[i]=g[i][j];
}
for(int x=1;x<=mn;x++){
for(int i=1;i<=hn;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=1;i<=hn;i++){
if(!visy[i]&&d>slack[i]) d=slack[i];
}
for(int i=1;i<=mn;i++){
if(visx[i]) lx[i]-=d;
}
for(int i=1;i<=hn;i++){
if(visy[i]) ly[i]+=d;
else slack[i]-=d;
}
}
}
int result = 0;
for(int i = 1; i <=mn; i++)
if(match[i]>-1)
result += g[match[i]][i];
return -result;
}
int main(){
while(scanf("%d%d",&n,&m)!=EOF,n+m){
for(int i=0;i<n;i++) scanf("%s",map[i]);
mn=hn=0;
for(int i=0;i<n;i++)
for(int j=0;j<m;j++){
if(map[i][j]=='m'){
man[++mn].x=i;
man[mn].y=j;
}
else if(map[i][j]=='H'){
house[++hn].x=i;
house[hn].y=j;
}
}
for(int i=1;i<=mn;i++)
for(int j=1;j<=hn;j++){
g[i][j]= - dit(man[i],house[j]);
}
printf("%d\n",KM());
}
return 0;
}