poj 2516(最小费用流)

#include <iostream>
#include <cstdio>
#include <queue>
#include <cmath>
#include <algorithm>
#include <cstring>
using namespace std;

#define inf 0x3f3f3f3f

int supply[55],demand[55];//某种商品的提供量和需求量
int dis[110],pre[110],s,t;//s为超级源点，t为超级汇点
int order[55][55];//第i个商店对第j种物品的需求
int store[55][55];//第i个供应站对第j个物品的存储量
int flow[110][110],cap[110][110];//边的流量,容量
int cost[55][55][55];//cost[k][j][i]第k种商品从第j个供应站运到第i个商店的花费
int kcost[110][110];//每件商品在某一条边的花费。从供应站到商店方向
int vis[110];
int n,m,k;

int spfa()
{
    queue<int>q;
    memset(vis,0,sizeof(vis));
    memset(pre,-1,sizeof(pre));
    q.push(s);
    fill(&dis[1],&dis[n+m+2],inf);//dis[1----n+m+1]==inf
    dis[s]=0;
    vis[s]=1;
    while(!q.empty())
    {
        int cur=q.front();q.pop();
        vis[cur]=0;
        for(int i=0;i<=n+m+1;i++)
        {
            if(cap[cur][i] >flow[cur][i] && dis[i] > dis[cur] + kcost[cur][i])
            {
                dis[i]=dis[cur]+kcost[cur][i];
                pre[i]=cur;
                if(!vis[i])
                {
                    vis[i]=1;
                    q.push(i);
                }
            }
        }
    }
    return pre[t]!=-1;//pre[t]==-1表示无增广路
}

void costflow()
{
    int minf=inf;
    while(spfa())
    {
        for(int u=t;pre[u] != -1;u=pre[u])
        {
            minf=min(minf,cap[pre[u]][u]-flow[pre[u]][u]);
        }
        for(int u=t;pre[u] !=-1 ;u=pre[u])//沿着最小费用路进行增广
        {
            flow[pre[u]][u]+=minf;
            flow[u][pre[u]]-=minf;
        }
    }
}

int main()
{
    while(scanf("%d%d%d",&n,&m,&k))
    {
        if(!n && !m && !k) break;
        int flag=1;//判断是否无解
        memset(supply,0,sizeof(supply));
        memset(demand,0,sizeof(demand));
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=k;j++)
            {
                scanf("%d",&order[i][j]);
                demand[j]+=order[i][j];
            }
        }
        for(int i=1;i<=m;i++)
        {
            for(int j=1;j<=k;j++)
            {
                scanf("%d",&store[i][j]);
                supply[j]+=store[i][j];
            }
        }
        for(int p=1;p<=k;p++)
        {
            for(int i=1;i<=n;i++)
            {
                for(int j=1;j<=m;j++)
                {
                    scanf("%d",&cost[p][j][i]);
                }
            }
        }
        for(int i=1;i<=k;i++)
        {
            if(demand[i]>supply[i])
            {
                flag=0;
                break;
            }
        }
        int flowcost=-1;
        if(flag)
        {
            s=0;
            t=n+m+1;
            //按照商品数构图k次
            //cap,kcost，下标1----m为供应站,m+1--m+n为商店
            flowcost=0;
            for(int p=1;p<=k;p++)
            {
                memset(cap,0,sizeof(cap));
                memset(kcost,0,sizeof(kcost));
                memset(flow,0,sizeof(flow));
                for(int i=1;i<=m;i++)//源点到供应站建边
                    cap[s][i]=store[i][p];
                for(int i=1;i<=n;i++)
                    cap[i+m][t]=order[i][p];
                for(int i=1;i<=m;i++)
                {
                    for(int j=1;j<=n;j++)
                    {
                        cap[i][j+m]=store[i][p];
                        kcost[i][j+m]=cost[p][i][j];
                        kcost[j+m][i]=-cost[p][i][j];//负费用,表示回流费用增加
                    }
                }
                costflow();
                for(int i=1;i<=m;i++)
                {
                    for(int j=1;j<=n;j++)
                        flowcost+=kcost[i][j+m]*flow[i][j+m];//单价乘流量
                }
            }
        }
        printf("%d\n",flowcost);
    }
    return 0;
}