/*
给你一个N行M列的矩阵,其中“.”代表空地,“H”代表房子,“m”代表人,其中有n个房子和n个人。
现在要求每个人进入一间房子,且人走一步需要支付1美元。
求最小需要花费多少美元才能让所有人都进入到房子中(每个人只能进入一间房子,每个房子只能容纳一个人)。
解题思路:
这道题其实就是二分图最优匹配的变形而已。
因为要求的其实是最小权值之和。而KM算法求的是最大权值之和。
在建图的时候,将每条边的权值变为负数。然后lx[i]初始化为-INT_MAX,结果输出-ans,就可以得到最小权值。
*/
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
const int MAXNODE = 105;
typedef int Type;
const Type INF = 0x3f3f3f3f;
struct KM {
int n;
Type g[MAXNODE][MAXNODE];
Type Lx[MAXNODE], Ly[MAXNODE], slack[MAXNODE];
int left[MAXNODE];
bool S[MAXNODE], T[MAXNODE];
void init(int n) {
this->n = n;
}
void add_Edge(int u, int v, Type val) {
g[u][v] = val;
}
bool dfs(int i) {
S[i] = true;
for (int j = 0; j < n; j++) {
if (T[j]) continue;
Type tmp = Lx[i] + Ly[j] - g[i][j];
if (!tmp) {
T[j] = true;
if (left[j] == -1 || dfs(left[j])) {
left[j] = i;
return true;
}
} else slack[j] = min(slack[j], tmp);
}
return false;
}
void update() {
Type a = INF;
for (int i = 0; i < n; i++)
if (!T[i]) a = min(a, slack[i]);
for (int i = 0; i < n; i++) {
if (S[i]) Lx[i] -= a;
if (T[i]) Ly[i] += a;
}
}
int km() {
for (int i = 0; i < n; i++) {
left[i] = -1;
Lx[i] = -INF; Ly[i] = 0;
for (int j = 0; j < n; j++)
Lx[i] = max(Lx[i], g[i][j]);
}
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) slack[j] = INF;
while (1) {
for (int j = 0; j < n; j++) S[j] = T[j] = false;
if (dfs(i)) break;
else update();
}
}
int ans = 0;
for (int i = 0; i < n; i++)
ans += g[left[i]][i];
return ans;
}
} gao;
const int N = 105;
int n, m;
char str[N];
struct Point {
int x, y;
Point() {}
Point(int x, int y) {
this->x = x;
this->y = y;
}
} hp[N],mp[N];
int dis(Point a, Point b) {
return abs(a.x - b.x) + abs(a.y - b.y);
}
int hn, mn;
int main() {
while (~scanf("%d%d", &n, &m) && n || m) {
hn = mn = 0;
for (int i = 0; i < n; i++) {
scanf("%s", str);
for (int j = 0; j < m; j++) {
if (str[j] == 'H') hp[hn++] = Point(i, j);
if (str[j] == 'm') mp[mn++] = Point(i, j);
}
}
gao.n = hn;
for (int i = 0; i < hn; i++) {
for (int j = 0; j < mn; j++) {
gao.g[i][j] = -dis(hp[i], mp[j]);
}
}
printf("%d\n", -gao.km());
}
return 0;
}
HDU Going Home(KM算法)
最新推荐文章于 2019-08-28 19:34:41 发布