Minimum Cost-最小费用最大流

题目描述

之前还自以为全部塞在一个图里面就可以搞了。谁知，这样不行。
分开搞网络流，一方面可以节约某些时间，一方面有时候会简化问题。

#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
#define MAXPT 126000
#define MAXN 50
#define INF 2000000000

struct node{
    int v,w,c;
    node *next,*back;
}edge[MAXPT*4+10],*adj[MAXPT+10],*ecnt=&edge[0],*fa[MAXPT+10];

int n,m,k,need[MAXN+10][MAXN+10],have[MAXN+10][MAXN+10],cost[MAXN+10][MAXN+10][MAXN+10];
int s,t,dist[MAXPT+10],ans,tot;
bool vis[MAXPT+10];

inline void Init()
{
    tot=0;
    ecnt=&edge[0];
    //memset(edge,0,sizeof edge); //加上这个瞬间慢成什么了。。
    memset(adj,0,sizeof adj);
}
void addedge(int u,int v,int c,int w)
{
    node *p=++ecnt;
    p->v=v;
    p->c=c,p->w=w;
    p->next=adj[u],p->back=ecnt+1;
    adj[u]=p;
    p=++ecnt;
    p->v=u;
    p->c=0,p->w=0;
    p->next=adj[v],p->back=ecnt-1;
    adj[v]=p;
}
void read()
{
    for(int i=1;i<=n;i++)
        for(int j=1;j<=k;j++)
            scanf("%d",&need[i][j]);
    for(int i=1;i<=m;i++)
        for(int j=1;j<=k;j++)
            scanf("%d",&have[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][i][j]);
}
bool SPFA()
{
    int u,v;
    memset(vis,0,sizeof vis);
    for(int i=s;i<=t;i++)
        dist[i]=INF;
    queue<int> que;
    que.push(s);
    vis[s]=true,dist[s]=0;
    while(!que.empty()){
        u=que.front(); que.pop();
        vis[u]=false;
        for(node *p=adj[u];p;p=p->next){
            v=p->v;
            if(p->c&&dist[v]>dist[u]+p->w){
                dist[v]=dist[u]+p->w;
                fa[v]=p;
                if(!vis[v]){
                    que.push(v);
                    vis[v]=true;
                }
            }
        }
    }
    if(dist[t]==INF) return false;
    else return true;
}
bool mincost_maxflow()
{
    int flow=0;
    while(SPFA()){
        int delta=INF;
        node *p;
        for(int u=t;u!=s;u=p->back->v){
            p=fa[u];
            delta=min(delta,p->c);
        }
        flow+=delta;
        for(int u=t;u!=s;u=p->back->v){
            p=fa[u];
            p->c-=delta;
            p->back->c+=delta;
            p->back->w=-p->w;
            ans+=delta*p->w;
        }
    }
    if(tot==flow) return true;
    else return false;
}
bool workout(int kth)
{
    Init();
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            addedge(i,n+j,INF,cost[kth][i][j]);
    s=0,t=n+m+1;
    for(int i=1;i<=n;i++){
        addedge(s,i,need[i][kth],0);
        tot+=need[i][kth];
    }
    for(int i=1;i<=m;i++)
        addedge(n+i,t,have[i][kth],0);
    return mincost_maxflow();
}
int main()
{
    while(scanf("%d%d%d",&n,&m,&k)){
        if(!n&&!m&&!k) break;
        read();
        ans=0;
        for(int i=1;i<=k;i++)
            if(!workout(i)){
                ans=-1;
                break;
            }
        printf("%d\n",ans);
    }
}

转载于:https://www.cnblogs.com/katarinayuan/p/6572855.html