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 = 1000 + 5;

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 = 0; 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, 0, cost));
        edges.push_back(Edge(to, from, 0, 0, -cost));
        m = edges.size();
        G[from].push_back(m - 2);
        G[to].push_back(m - 1);
    }

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

        queue<int> Q;
        Q.push(s);
        while (!Q.empty()){
            int u = Q.front(); Q.pop();
            inq[u] = 0;
            for (int i = 0; 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] = 1; }
                }
            }
        }

        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] ^ 1].flow -= a[t];
        }
        return true;
    }

    int MincostMaxdflow(int s, int t){
        int flow = 0, cost = 0;
        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=0, cnt_h=0;
        for(int i=0;i<n;i++)
        {
            for(int j=0;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=0, dst=2*n+1;
        t.init(dst+1);

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

        for(int i=1;i<=cnt_p;i++)
        {
            for(int j=1;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,1,dis);
            }
        }

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