POJ2516 构图+k次费用流

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

题目大意及分析：

　　http://blog.csdn.net/lyy289065406/article/details/6742534

　　这个题解非常非常详细，我的构图和他反了下，本质一样。

　　这题KM的做法： http://blog.csdn.net/ChinaCzy/article/details/5813011

P.S. 这题我错了三天，三天来一直在搞这题，把网上一份和我的类似的代码找来改成我的风格，不断差错……

　　不觉得把时间这样浪费是一种对生命的亵渎吗？区域赛说近也近了，三天可以做多少事啊！

1、我过去习惯的spfa的写法都是错误的，以后一律使用循环队列，且用 if( head==maxn ) head= 0; 更快！本题边权非负，所以不需要开cnt[]来记录是否有负环，既然是循环队列那么必须要用 while( head != tail )来结束循环而不是习惯性地 head<tail；

2、所谓费用流，是： 费用*流量，我忘记这一点就错了很久……

3、一定记得反边的边权是正向边边权的相反数；

4、这题开始我一直想不通那个k个物品怎么办，后来是看了题解才知道他们确实是独立的，要分别做费用流，可是我明明会这种费用流，为什么会错这么久呢？ 我实在是不知道错在哪里了。

代码：

/*2516    Accepted    404K    266MS    C++    2635B    2012-06-22 14:08:10*/
#include <cstdio>
#include <cstring>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

#define mpair make_pair
#define pii pair<int,int>
#define MM(a,b) memset(a,b,sizeof(a));
typedef long long lld;
typedef unsigned long long u64;
template<class T> bool up_max(T& a,const T& b){return b>a? a=b,1 : 0;}
template<class T> bool up_min(T& a,const T& b){return b<a? a=b,1 : 0;}
#define maxn 110
const int inf= 2100000000;

int n,m,k;
int ST, ED;
int c[maxn][maxn], f[maxn][maxn];

bool vis[maxn];
int que[maxn+10], dis[maxn], pre[maxn];
bool spfa(){
    for(int i=0;i<=ED;++i) dis[i]= inf, vis[i]= 0;
    int head= 0, tail= 0;
    que[tail++]= ST;
    vis[ST]= 1;
    dis[ST]= 0;
    while( head!=tail ){
        int u= que[head++];
        if( head==maxn ) head= 0;
        vis[u]= 0;
        for(int v=0;v<=ED;++v){
            if( f[u][v]>0 && up_min( dis[v], dis[u]+c[u][v] ) ){
                pre[v]= u;
                if( !vis[v] ){
                    que[tail++]= v;
                    if( tail==maxn ) tail= 0;
                    vis[v]= 1;
                }
            }
        }
    }
    return dis[ED]!=inf;
}

int mincost(){
    int ret= 0;
    while( spfa() ){
        int t= inf;
        for(int i=ED;i!=ST;i=pre[i])
            up_min( t, f[ pre[i] ][i] );
        for(int i=ED;i!=ST;i=pre[i]){
            f[ pre[i] ][i] -= t;
            f[i][ pre[i] ] += t;
        }
        ret+= t*dis[ED];
    }
    return ret;
}

#define N 55
int a[maxn][maxn], b[maxn][maxn];
int solve(){
    int i,j;
    for(i=1;i<=n;++i)for(j=1;j<=k;++j)scanf("%d", &a[i][j]);
    for(i=1;i<=m;++i)for(j=1;j<=k;++j)scanf("%d", &b[i][j]);

    bool flag= 0;
    for(int p=1;p<=k;++p){
        int s1= 0, s2= 0;
        for(i=1;i<=n;++i) s1+= a[i][p];
        for(i=1;i<=m;++i) s2+= b[i][p];
        if( s1>s2 ){
            flag= 1; break;
        }
    }

    ST= 0, ED= n+m+1;
    int cost= 0;
    MM( c, 0 );
    for(int p=1;p<=k;++p){
        for(i=1;i<=n;++i)
            for(j=1;j<=m;++j){
                scanf("%d", &c[i][n+j]);
                c[n+j][i]= -c[i][n+j];
            }
        if( flag ) continue;

        MM( f, 0 );
        for(i=1;i<=n;++i) f[ST][i]= a[i][p];
        for(i=1;i<=m;++i) f[n+i][ED]= b[i][p];
        for(i=1;i<=n;++i)for(j=1;j<=m;++j)f[i][n+j]= inf;
        cost+= mincost();
    }
    if( flag ) cost= -1;
    return cost;
}

int main()
{
    //freopen("poj2516.in","r",stdin);
    while( cin>>n>>m>>k, n+m+k )
        cout<< solve() <<endl;
}

/*2516    Accepted    400K    297MS    C++    2562B    2012-06-22 14:16:59*/
#include <cstdio>
#include <cstring>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

#define mpair make_pair
#define pii pair<int,int>
#define MM(a,b) memset(a,b,sizeof(a));
typedef long long lld;
typedef unsigned long long u64;
template<class T> bool up_max(T& a,const T& b){return b>a? a=b,1 : 0;}
template<class T> bool up_min(T& a,const T& b){return b<a? a=b,1 : 0;}
#define maxn 110
const int inf= 2100000000;

int n,m,k;
int ST, ED;
int c[maxn][maxn], f[maxn][maxn];

bool vis[maxn];
int que[maxn+10], dis[maxn], pre[maxn];
bool spfa(){
    for(int i=0;i<=ED;++i) dis[i]= inf, vis[i]= 0;
    int head= 0, tail= 0;
    que[tail++]= ST;
    vis[ST]= 1;
    dis[ST]= 0;
    while( head!=tail ){
        int u= que[head++];
        if( head==maxn ) head= 0;
        vis[u]= 0;
        for(int v=0;v<=ED;++v){
            if( f[u][v]>0 && up_min( dis[v], dis[u]+c[u][v] ) ){
                pre[v]= u;
                if( !vis[v] ){
                    que[tail++]= v;
                    if( tail==maxn ) tail= 0;
                    vis[v]= 1;
                }
            }
        }
    }
    return dis[ED]!=inf;
}

pii mincost(){
    int flow= 0, cost= 0;
    while( spfa() ){
        int t= inf;
        for(int i=ED;i!=ST;i=pre[i])
            up_min( t, f[ pre[i] ][i] );
        flow+= t;
        for(int i=ED;i!=ST;i=pre[i]){
            f[ pre[i] ][i] -= t;
            f[i][ pre[i] ] += t;
        }
        cost+= t*dis[ED];
    }
    return pii( flow, cost );
}

#define N 55
int a[maxn][maxn], b[maxn][maxn];
int solve(){
    ST= 0, ED= n+m+1;
    int i,j,cost= 0;
    for(i=1;i<=n;++i)for(j=1;j<=k;++j)scanf("%d", &a[i][j]);
    for(i=1;i<=m;++i)for(j=1;j<=k;++j)scanf("%d", &b[i][j]);

    bool flag= 0;
    MM( c, 0 );
    for(int p=1;p<=k;++p){
        MM( f, 0 );
        int ss= 0;
        for(i=1;i<=n;++i){
            ss+= a[i][p];
            for(j=1;j<=m;++j){
                scanf("%d", &c[i][n+j]);
                c[n+j][i]= -c[i][n+j];
                f[i][n+j]= inf;
            }
        }
        if( flag ) continue;

        for(i=1;i<=n;++i) f[ST][i]= a[i][p];
        for(i=1;i<=m;++i) f[n+i][ED]= b[i][p];
        pii tmp= mincost();
        cost+= tmp.second;
        if( ss!=tmp.first ) flag= 1;
    }
    if( flag ) cost= -1;
    return cost;
}

int main()
{
    //freopen("poj2516.in","r",stdin);
    while( cin>>n>>m>>k, n+m+k )
        cout<< solve() <<endl;
}