http://poj.org/problem?id=2195
题意:一个n*m的矩阵,其中每个'm'代表一个人,每个‘H'代表一个房子,且人和房子的数目相同,要求每个人找到一个属于自己的房子,每个房子只能住一个人,人到房子的花费就是它们之间的曼哈顿距离。问他们的最小花费。
思路:和 Minimum Cost 一样,貌似这个更简单点,因为只需计算一次最小费用最大流。
附加一个超级源点S和超级汇点T。
s 指向所有的人,容量为1,费用为0; 每个人指向所有的房子,容量为INF,费用为人和房子的曼哈顿距离; 所有房子指向t,容量为1,费用为0。 从源点到汇点求费用流。
#include <stdio.h> #include <string.h> #include <queue> #include <algorithm> #include <cmath> using namespace std; const int maxn = 110; const int INF = 0x3f3f3f3f; struct node { int x,y; }house[maxn],man[maxn]; int n,m; int hh,mm,num; int s,t; int cost[maxn*2][maxn*2],cap[maxn*2][maxn*2]; int mincost; queue <int> que; int inque[maxn*2],pre[maxn*2],dis[maxn*2]; bool spfa() { while(!que.empty()) que.pop(); memset(inque,0,sizeof(inque)); memset(pre,-1,sizeof(pre)); memset(dis,INF,sizeof(dis)); dis[s] = 0; inque[s] = 1; que.push(s); while(!que.empty()) { int u = que.front(); que.pop(); inque[u] = 0; for(int v = s; v <= t; v++) { if(cap[u][v] && dis[v] > dis[u]+cost[u][v]) { dis[v] = dis[u]+cost[u][v]; pre[v] = u; if(!inque[v]) { inque[v] = 1; que.push(v); } } } } if(pre[t] == -1) return false; return true; } void MCMF() { while(spfa()) { for(int u = t; u != s; u = pre[u]) { cap[pre[u]][u] -= 1; cap[u][pre[u]] += 1; } mincost += dis[t]; } } int main() { char str[maxn]; while(~scanf("%d %d",&n,&m)) { if(n == 0 && m == 0) break; hh = 0; mm = 0; for(int i = 1; i <= n; i++) { scanf("%s",str+1); for(int j = 1; j <= m; j++) { if(str[j] == 'H') house[++hh] = (struct node){i,j}; if(str[j] == 'm') man[++mm] = (struct node){i,j}; } } s = 0; t = mm+hh+1; memset(cost,0,sizeof(cost)); memset(cap,0,sizeof(cap)); for(int i = 1; i <= mm; i++) { for(int j = 1; j <= hh; j++) { int tmp = abs(man[i].x-house[j].x) + abs(man[i].y-house[j].y); cost[i][j+mm] = tmp; cost[j+mm][i] = -tmp; cap[i][j+mm] = INF; } } for(int i = 1; i <= mm; i++) { cap[s][i] = 1; cap[i+mm][t] = 1; } mincost = 0; MCMF(); printf("%d\n",mincost); } return 0; }