POJ【2516】Minimum Cost（最小费用流）

题目链接：http://poj.org/problem?id=2516
分析
这道题考查最小费用流，首先建立源点和汇点，源点指向所有的供应商，边的流量为供应商的供应量，费用为0，商店指向汇点，边的流量为商店的需求量，费用为0，供应商指向商店，流量为无穷大，代表流量没有限制（前提是供过于或等于求），费用为运送商品的费用。注意这道题需要分k次求最大流，这样每一次建边，边的流量就是每种物品的供应量或需求量。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int INF = 0x3f3f3f3f;
const int N = 300 + 10;
int n,m,K;
int from[N][N],to[N][N],cost[N][N],need[N],supply[N];
int a[N],d[N],inq[N];
int s,t;
struct Edge {
    int to,next,w,c;
} e[N * N];
struct Pre {
    int pre,id;
} p[N];
int head[N],cnt;
void add(int u,int v,int w,int c)
{
    e[cnt] = (Edge){v,head[u],w,c};
    head[u] = cnt++;
    e[cnt] = (Edge){u,head[v],0,-c};
    head[v] = cnt++;
}
bool SPFA(int& maxflow,int& mincost)//注意这里是引用
{
    memset(d,0x3f,sizeof(d));
    memset(inq,0,sizeof(inq));
    memset(p,-1,sizeof(p));
    queue<int> q;
    q.push(s);
    d[s] = 0;
    a[s] = INF;
    p[s].pre = s;
    while(!q.empty()) {
        int u = q.front();q.pop();
        inq[u] = 0;
        for(int i = head[u]; i != -1; i = e[i].next) {
            int v = e[i].to;
            int c = e[i].c;
            if(d[v] > d[u] + c && e[i].w > 0) {
                d[v] = d[u] + c;//最小费用
                p[v].pre = u;
                p[v].id = i;
                a[v] = min(a[u],e[i].w);//最大流
                if(!inq[v]) {
                    q.push(v);
                    inq[v] = 1;
                }
            }
        }
    }
    if(d[t] == INF) return false;//如果抵达不了终点
    maxflow += a[t];
    mincost += d[t] * a[t];
    int u = t;
    while(u != s) {
        e[p[u].id].w -= a[t];
        e[p[u].id ^ 1].w += a[t];
        u = p[u].pre; 
    }
    return true;
}
int minCost()
{
    int maxflow = 0,mincost = 0;
    while(SPFA(maxflow,mincost));
    return mincost;
}
int main()
{
    while(scanf("%d%d%d",&n,&m,&K) && n + m + K) {
        int flag = 1;
        memset(need,0,sizeof(need));
        memset(supply,0,sizeof(supply));
        for(int i = 0; i < n; i++)
            for(int j = 0; j < K; j++) {
                scanf("%d",&to[i][j]);
                need[j] += to[i][j];
            }
        for(int i = 0; i < m; i++)
            for(int j = 0; j < K; j++) {
                scanf("%d",&from[i][j]);
                supply[j] += from[i][j];
            }
        for(int i = 0; i < K; i++)
            if(supply[i] < need[i]) {//如果出现供不应求的情况
                flag = 0;
                break;
            }
        int ans = 0;
        s = n + m,t = s + 1;//建立一个源点和汇点
        for(int i = 0; i < K; i++) {
            for(int j = 0; j < n; j++)
                for(int k = 0; k < m; k++)
                    scanf("%d",&cost[j][k]);
            if(flag) {//如果供过于求或者供等于求
                memset(head,-1,sizeof(head));
                cnt = 0;
                for(int j = 0; j < m; j++)
                    add(s,j,from[j][i],0);//源点指向供应商，流量为对应商品的供应量
                for(int j = 0; j < n; j++)
                    add(j + m,t,to[j][i],0);//商店指向汇点，流量为对应商品的需求量
                for(int j = 0; j < n; j++)
                    for(int k = 0; k < m; k++)
                        add(k,j + m,INF,cost[j][k]);//INF可以理解为没有限制流量
                ans += minCost();
            }
        }
        if(flag) printf("%d\n",ans);
        else printf("-1\n");
    }
    return 0;
}