POJ 2195 Going Home（费用流）

http://poj.org/problem?id=2195

题意：

在一个网格地图上，有n个小人和n栋房子。在每个时间单位内，每个小人可以往水平方向或垂直方向上移动一步，走到相邻的方格中。对每个小人，每走一步需要支付1美元，直到他走入到一栋房子里。每栋房子只能容纳一个小人。

计算出让n个小人移动到n个不同的房子需要支付的最小费用。

思路：

源点和每个人相连，容量为1，费用为0。

汇点和每栋房子相连，容量为1，费用为0。

每个人和每栋房子相连，容量为1，费用为人和房子之间的距离。

这样一来，跑一遍费用流即可。

 #include<iostream>
 #include<algorithm>
 #include<cstring>
 #include<cstdio>
 #include<sstream>
 #include<vector>
 #include<stack>
 #include<queue>
 #include<cmath>
 #include<map>
 #include<set>
 using namespace std;
 typedef long long ll;
 typedef long long ull;
 typedef pair<int,int> pll;
 const int INF = 0x3f3f3f3f;
 const int maxn =  + ;

 int n, m, k;

 struct Edge
 {
     int from, to, cap, flow, cost;
     Edge(int u, int v, int c, int f, int w) :from(u), to(v), cap(c), flow(f), cost(w) {}
 };

 struct MCMF
 {
     int n, m;
     vector<Edge> edges;
     vector<int> G[maxn];
     int inq[maxn];
     int d[maxn];
     int p[maxn];
     int a[maxn];

     void init(int n)
     {
         this->n = n;
         for (int i = ; i<n; i++) G[i].clear();
         edges.clear();
     }

     void AddEdge(int from, int to, int cap, int cost)
     {
         edges.push_back(Edge(from, to, cap, , cost));
         edges.push_back(Edge(to, from, , , -cost));
         m = edges.size();
         G[from].push_back(m - );
         G[to].push_back(m - );
     }

     bool BellmanFord(int s, int t, int &flow, int & cost)
     {
         for (int i = ; i<n; i++) d[i] = INF;
         memset(inq, , sizeof(inq));
         d[s] = ; inq[s] = ; p[s] = ; a[s] = INF;

         queue<int> Q;
         Q.push(s);
         while (!Q.empty()){
             int u = Q.front(); Q.pop();
             inq[u] = ;
             for (int i = ; i<G[u].size(); i++){
                 Edge& e = edges[G[u][i]];
                 if (e.cap>e.flow && d[e.to]>d[u] + e.cost){
                     d[e.to] = d[u] + e.cost;
                     p[e.to] = G[u][i];
                     a[e.to] = min(a[u], e.cap - e.flow);
                     if (!inq[e.to]) { Q.push(e.to); inq[e.to] = ; }
                 }
             }
         }

         if (d[t] == INF) return false;
         flow += a[t];
         cost += d[t] * a[t];
         for (int u = t; u != s; u = edges[p[u]].from)
         {
             edges[p[u]].flow += a[t];
             edges[p[u] ^ ].flow -= a[t];
         }
         return true;
     }

     int MincostMaxdflow(int s, int t){
         int flow = , cost = ;
         while (BellmanFord(s, t, flow, cost));
         return cost;
     }
 }t;

 struct node
 {
     int x, y;
 }people[maxn],house[maxn];

 int main()
 {
     //freopen("in.txt","r",stdin);
     while(~scanf("%d%d",&n,&m) && n && m)
     {
         char c;
         int cnt_p=, cnt_h=;
         for(int i=;i<n;i++)
         {
             for(int j=;j<m;j++)
             {
                 cin>>c;
                 if(c=='H')       {house[++cnt_h].x=i;house[cnt_h].y=j;}
                 else if(c=='m')  {people[++cnt_p].x=i;people[cnt_p].y=j;}
             }
         }

         int n=cnt_h;
         int src=, dst=*n+;
         t.init(dst+);

         for(int i=;i<=cnt_p;i++)  t.AddEdge(src,i,,);
         for(int i=;i<=cnt_h;i++)  t.AddEdge(n+i,dst,,);

         for(int i=;i<=cnt_p;i++)
         {
             for(int j=;j<=cnt_h;j++)
             {
                 int dis=abs(people[i].x-house[j].x)+abs(people[i].y-house[j].y);
                 t.AddEdge(i,n+j,,dis);
             }
         }

         printf("%d\n",t.MincostMaxdflow(src,dst));
     }
     return ;
 }