poj 2516Minimum Cost 最小费用最大流

//poj 2516
//poj 2516
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAX=0xfffffff;
int N,M,K;
int S,T;
const int Max=300;
int cost[Max][Max];//花费
int cap[Max][Max];//容量
//path[]最短路的前继定点，dist[]起点到此点的距离，vis标记是否访问过,q循环队列
int dist[Max];
int path[Max];
bool visit[Max];
int q[Max*10];
int minCost;
bool spfa( )
{
    memset(path,-1,sizeof(path));
    memset(visit,0,sizeof(visit));
    for(int i=0;i<=T;i++)   dist[i]=MAX;    //无法用memset赋值，memset(dist,MAX,sizeof(dist));
    dist[S]=0;
    int num=0,cnt=0;
    q[num++]=S;
    while(cnt<num)
    {
        int temp=q[cnt++];    //出队时，前指针+1，入队时，后指针+1
        visit[temp]=0;
        for(int i=0;i<=T;i++)
        {
            if(cap[temp][i]&&dist[temp]+cost[temp][i]<dist[i])
            {
                path[i]=temp;
                dist[i]=dist[temp]+cost[temp][i];
                if(!visit[i])
                {
                    q[num++]=i;
                    visit[i]=1;
                }
            }
        }
    }
    return dist[T]<MAX;
}
int MCMF()
{
    minCost=0;
    while(spfa())   //增广找最大流,和bfs（）的差别在于找的是最短路，其他一样
    {
        int maxFlow=MAX;
        int pre=T;     //从根节点向前递推
        while(path[pre]!=-1)
        {
            maxFlow=min(maxFlow,cap[path[pre]][pre]);
            pre=path[pre];
        }
        pre=T;
        minCost+=dist[T]*maxFlow;
        while(path[pre]!=-1)      //更新增广路径上的边
        {
            cap[path[pre]][pre]-=maxFlow;
            cap[pre][path[pre]]+=maxFlow;
            pre=path[pre];
        }
    }
    return minCost;
}
int main( )
{
    //freopen("1.txt","r",stdin);
    while(scanf("%d %d %d",&N,&M,&K)&&N)
    {
        int ans=0;
        int need[60][60];
        int needK[60];
        memset(needK,0,sizeof(needK));
        for(int i=1;i<=N;i++)
            for(int j=1;j<=K;j++)
            {
                scanf("%d",&need[i][j]);
                needK[j]+=need[i][j];
            }
        int have[60][60];
        int haveK[60];
        memset(haveK,0,sizeof(haveK));
        for(int i=1;i<=M;i++)
            for(int j=1;j<=K;j++)
            {
                scanf("%d",&have[i][j]);
                haveK[j]+=have[i][j];
            }
        bool flag=1;
        for(int i=1;i<=K;i++)
            if(haveK[i]<needK[i])
            {
                flag=0;
                break;
            }
        T=N+M+1;
        S=0;
        for(int k=1;k<=K;k++)
        {
            memset(cap,0,sizeof(cap));
            for(int i=1;i<=N;i++)
                for(int j=1;j<=M;j++)
                {
                    scanf("%d",&cost[j][M+i]);
                    cost[M+i][j]=-cost[j][M+i];
                    cap[j][M+i]=MAX;
                }
            if(!flag)  continue;
            for(int i=1;i<=M;i++)
            {
                cap[0][i]=have[i][k];
                cost[0][i]=cost [i][0]=0;
            }
            for(int i=1;i<=N;i++)
            {
                cap[M+i][T]=need[i][k];
                cost[M+i][T]=cost[T][M+i]=0;
            }
            ans+=MCMF();
        }
        if(!flag)  printf("-1\n");
        else  printf("%d\n",ans);
    }
    return 0;
}