poj 2516 Minimum Cost【最大流最小费用流】

题目链接：http://poj.org/problem?id=2516；

题目大意：有 n 个商店，m 个仓库，k 种商品，问满足供货商要求的最小花费是多少，如果不能满足输出-1；前 n 行为每个商店需要商品的数量，接着 m 行为每个仓库能放的多少个每种商品，然后是 k 个 n*m 的矩阵，表示第 k 个物品从仓库 i 到商店 j 的花费：

思路：每种商品求一下最小费用流，求k次；

我的代码源点是 0 ，汇点是 n+m+1，与源点连的是商店，与汇点相连的是仓库；

#include<cstring>
#include<string>
#include<cstdio>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<map>
#include<vector>
#include<stack>
#define inf 0x3f3f3f3f
#include<queue>
#include<set>
using namespace std;
typedef long long ll;
const int N=105;
const int M=20005;

//建图
struct node
{
    int v,f,cos,ne;
} edge[M];
int head[N*2],pre[N*2],dist[N*2],vis[N*2];
int s,t,e;

//读数据
int ned[N][N],has[N][N],val[N][N][N];
int nsum[N],hsum[N];

void add_insert(int a,int b,int c,int d)
{
    edge[e].v=b;
    edge[e].f=c;
    edge[e].cos=d;
    edge[e].ne=head[a];
    head[a]=e++;
}

void add(int a,int b,int c,int d)
{
    add_insert(a,b,c,d);
    add_insert(b,a,0,-d);
}

int spaf()
{
    memset(pre,-1,sizeof(pre));
    memset(dist,inf,sizeof(dist));
    memset(vis,0,sizeof(vis));
    queue<int>que;
    vis[s]=1;
    dist[s]=0;
    que.push(s);
    while(!que.empty())
    {
        int u=que.front();
        que.pop();
        vis[u]=0;
        for(int i=head[u]; i!=-1; i=edge[i].ne)
        {
            int v=edge[i].v,f=edge[i].f,c=edge[i].cos;
            if(dist[v]>dist[u]+c&&f)
            {
                dist[v]=dist[u]+c;
                pre[v]=i;
                if(!vis[v])
                {
                    vis[v]=1;
                    que.push(v);
                }
            }
        }
    }
    return pre[t]!=-1;
}

int MCMF()
{
    int flow=0,cost=0;
    while(spaf())
    {
        int mini=inf;
        for(int i=pre[t]; i!=-1; i=pre[edge[i^1].v])
            mini=min(mini,edge[i].f);
        for(int i=pre[t]; i!=-1; i=pre[edge[i^1].v])
        {
            edge[i].f-=mini;
            edge[i^1].f+=mini;
            cost+=edge[i].cos*mini;
        }
        flow+=mini;
    }
    return cost;
}

int main()
{
    int n,m,k;
    while(~scanf("%d %d %d",&n,&m,&k))
    {
        if(!n&&!m&&!k) break;
        s=0;t=n+m+1;
        memset(nsum,0,sizeof(nsum));
        memset(hsum,0,sizeof(hsum));
        int ans=0;
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=k;j++)
            {
                scanf("%d",&ned[i][j]);//记录商店需要的货物的数量
                nsum[j]+=ned[i][j];
            }
        }
        for(int i=1;i<=m;i++)
        {
            for(int j=1;j<=k;j++)
            {
                scanf("%d",&has[i][j]);//记录仓库存放的货物的数量
                hsum[j]+=has[i][j];
            }
        }
        for(int i=1;i<=k;i++)
        {
            for(int j=1;j<=n;j++)
            {
                for(int l=1;l<=m;l++)
                    scanf("%d",&val[i][j][l]);//记录第l种货物从i到j的花费
            }
        }
        int flag=1;
        for(int i=1;i<=k;i++)
        {
            if(nsum[i]>hsum[i])
            {
                flag=0;
                break;
            }
        }
        if(!flag)
        {
            printf("-1\n");
            continue;
        }
        for(int l=1;l<=k;l++)
        {
            memset(head,-1,sizeof(head));
            e=0;
            //源点到商店
            for(int i=1;i<=n;i++)
                add(s,i,ned[i][l],0);
            //仓库到汇点
            for(int i=1;i<=m;i++)
                add(n+i,t,has[i][l],0);
            //商店到仓库
            for(int i=1;i<=n;i++)
                for(int j=1;j<=m;j++)
                add(i,n+j,min(ned[i][l],has[j][l]),val[l][i][j]);
            ans+=MCMF();
        }
        printf("%d\n",ans);
    }
    return 0;
}