POJ 2516 Minimal Cost(最小费用最大流)

题意:

有n个商店,m个供给站,k种物资,每一个供给站运行每种物资到每一个商店
的价格都是不同的。首先每个商店对每种物资的需求也不同,让你安排一种方
法,使得满足要求的同事运费最少。
最小费用流的典型题目,但是有一个问题就是由于物品太多,如果一次性建一
个大图的话,不仅建图繁琐,而且以网络流的玄学复杂度很容易T,所以我们可
以对每一中物资进一个图跑一次最小费用流,最后相加就行了,因为每种物资
之间是互相不影响的。源点到每一个补给站连流量为物资数量,费用为0的边,
每一个补给站到每一个商店连一个流量为INF,费用为x的边,每一个商店到汇点
连一条流量为需求量,费用为0的边。
有一点要注意就是,你判断了无解之后,一定要把数据读完,不然就wa。。。
//
//  Created by  CQU_CST_WuErli
//  Copyright (c) 2015 CQU_CST_WuErli. All rights reserved.
//
// #include
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#define CLR(x) memset(x,0,sizeof(x))
#define OFF(x) memset(x,-1,sizeof(x))
#define MEM(x,a) memset((x),(a),sizeof(x))
#define For_UVa if (kase!=1) cout << endl
#define BUG cout << "I am here" << endl
#define lookln(x) cout << #x << "=" << x << endl
#define SI(a) scanf("%d",&a)
#define SII(a,b) scanf("%d%d",&a,&b)
#define SIII(a,b,c) scanf("%d%d%d",&a,&b,&c)
#define rep(flag,start,end) for(int flag=start;flag<=end;flag++)
#define Rep(flag,start,end) for(int flag=start;flag>=end;flag--)
#define Lson l,mid,rt<<1
#define Rson mid+1,r,rt<<1|1
#define Root 1,n,1
#define BigInteger bign
template <typename T> T gcd(const T& a,const T& b) {return b==0?a:gcd(b,a%b);}
const int MAX_L=2005;// For BigInteger
const int INF_INT=0x3f3f3f3f;
const long long INF_LL=0x7fffffff;
const int MOD=1e9+7;
const double eps=1e-9;
const double pi=acos(-1);
typedef long long  ll;
using namespace std;

const int N=1e5+100;
const int M=200;

int n,m,k;
int havekind[M],have[M][M];
int needkind[M],need[M][M]; 

int pnt[N],head[M],nxt[N],cap[N],cost[N],pre[N];
int cnt;

void add_edge(int u,int v,int f,int w) {
    pnt[cnt]=v;pre[cnt]=u;nxt[cnt]=head[u];head[u]=cnt;
    cap[cnt]=f;cost[cnt++]=w;
}

int a[M],p[M],vis[M];
ll d[M];

bool spfa(int s,int t,int &Flow,ll& Cost) {
    CLR(a);CLR(vis);
    rep(i,s,t) d[i]=INF_INT;
    queue<int> q;
    vis[s]=1;
    a[s]=INF_INT;
    d[s]=0;
    q.push(s);
    while (q.size()) {
        int x=q.front();q.pop();
        vis[x]=0;
        for (int i=head[x];~i;i=nxt[i]) {
            int v=pnt[i];
            if (d[v]>d[x]+cost[i] && cap[i]) {
                d[v]=d[x]+cost[i];
                p[v]=i;
                a[v]=min(a[x],cap[i]);
                if (!vis[v]) {
                    q.push(v);
                    vis[v]=1;
                }
            }
        }
    }
    if (d[t]>=INF_INT) return false;
    Flow+=a[t];
    Cost+=a[t]*d[t];
    for (int u=t;u!=s;u=pre[p[u]]) {
        cap[p[u]]-=a[t];
        cap[p[u]^1]+=a[t];
    }
    return true;
}

ll MCMF(int s,int t) {
    int Flow=0;
    ll Cost=0;
    while (spfa(s,t,Flow,Cost));
    return Cost;
}

int main(){
#ifdef LOCAL
    freopen("C:\\Users\\john\\Desktop\\in.txt","r",stdin);
//  freopen("C:\\Users\\john\\Desktop\\out.txt","w",stdout);
#endif
    while (SIII(n,m,k)==3 && !(!n && !m && !k)) {
        CLR(havekind);CLR(needkind);
        rep(i,1,n) rep(j,1,k) {
            SI(need[i][j]);
            needkind[j]+=need[i][j];            
        }
        rep(i,1,m) rep(j,1,k) {
            SI(have[i][j]);
            havekind[j]+=have[i][j];
        }
        int flag=1;
        rep(i,1,k) if (needkind[i]>havekind[i]) flag=0;
        int source=0,sink=m+n+1;
        ll ans=0;
        rep(z,1,k) {
            OFF(head);cnt=0;
            rep(i,1,n) rep(j,1,m) {
                int x;SI(x);
                add_edge(j,i+m,INF_INT,x);
                add_edge(i+m,j,0,-x);
            }
            if (!flag) continue;
            rep(i,1,m) {
                add_edge(source,i,have[i][z],0);
                add_edge(i,source,0,0);
            }
            rep(i,1,n) {
                add_edge(i+m,sink,need[i][z],0);
                add_edge(sink,i+m,0,0);
            }
            ans+=MCMF(source,sink);
        }
        if (!flag) cout << -1 << endl;
        else cout << ans << endl;
    }
    return 0;
}

你可能感兴趣的:(图论)