Hdu 1533（KM算法）

Hdu 1533

 

（1）思路：

KM算法模板题，

将每个人到家的距离求出来，
求出最小的距离，所以可以将距离变为负数，然后求出负数的最大值，就是正数的最小值。

 

（2）代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<cmath>
#include<utility>
using namespace std;
const int INF = 1e9+10;
vector <pair <int,int> > v1,v2;
char ss[120][120];
int vx[120],vy[120],lx[120],ly[120],n,pre[120],mp[120][120],cha;
int MAX(int x,int y){
	return x>y?x:y;
}
int MIN(int x,int y){
	return x<y?x:y;
}
int Dis(pair <int,int> t1,pair <int,int> t2){
	return abs(t1.first-t2.first)+abs(t1.second-t2.second);
}
bool dfs(int u){
	vx[u] = 1;
	for(int v=0;v<n;v++)
	if(!vy[v]&&lx[u]+ly[v]==mp[u][v]){
		vy[v] = 1;
		if(pre[v]==-1||dfs(pre[v])){
			pre[v] = u;
			return true;
		}
	}
	else if(!vy[v]){
		cha = MIN(cha,lx[u]+ly[v]-mp[u][v]);
	}
	return false;
}
int km(){
	memset(ly,0,sizeof(ly));
	for(int i=0;i<n;i++){
		lx[i] = mp[i][0];
		for(int j=1;j<n;j++) lx[i] = MAX(lx[i],mp[i][j]);
	}
	memset(pre,-1,sizeof(pre));
	for(int i=0;i<n;i++){
		while(true){
			memset(vx,0,sizeof(vx));
			memset(vy,0,sizeof(vy));
			cha = INF;
			if(dfs(i)) break;
			for(int j=0;j<n;j++){
				if(vx[j]==1) lx[j] -= cha;
				if(vy[j]==1) ly[j] += cha;
			}
		}
	}
	int ans = 0;
	for(int i=0;i<n;i++) ans += mp[pre[i]][i];
	return ans;
}
int main(void)
{
	int N,M;
	while(~scanf("%d%d",&N,&M)&&(N+M)){
		for(int i=1;i<=N;i++) scanf("%s",ss[i]+1);
		v1.clear();v2.clear();
		for(int i=1;i<=N;i++){
			for(int j=1;j<=M;j++)
			if(ss[i][j]=='m'){
				v1.push_back(make_pair(i,j));
			}else if(ss[i][j]=='H'){
				v2.push_back(make_pair(i,j)); 
			}
		}
		n = v1.size();
		//printf("%d -----",n);
		for(int i=0;i<n;i++){
			for(int j=0;j<n;j++){
				mp[i][j] = -Dis(v1[i],v2[j]);
			}
		}
		printf("%d\n",-km());
	}
	return 0;
}