运输问题（费用流）

运输问题

https://www.luogu.org/problemnew/show/P4015

题目描述

W 公司有 m 个仓库和 n 个零售商店。第 i 个仓库有 ai​ 个单位的货物；第 j 个零售商店需要 bj​ 个单位的货物。

货物供需平衡

从第 i 个仓库运送每单位货物到第 j个零售商店的费用为 cij​​​ 。

试设计一个将仓库中所有货物运送到零售商店的运输方案，使总运输费用最少。

输入输出格式

输入格式：

第 1 行有 2 个正整数 m 和 n，分别表示仓库数和零售商店数。

接下来的一行中有 m 个正整数 i​，表示第 i个仓库有 ai​个单位的货物。

再接下来的一行中有 n 个正整数 bj​，表示第 j个零售商店需要 bj​ 个单位的货物。

接下来的 m 行，每行有 n 个整数，表示从第 i 个仓库运送每单位货物到第 j 个零售商店的费用 cij​。

输出格式：

两行分别输出最小运输费用和最大运输费用。

输入输出样例

输入样例#1： 

2 3
220 280
170 120 210
77 39 105
150 186 122

输出样例#1： 

48500
69140

说明

1≤n,m≤100

费用分别用正的和负的建图，跑两次即可

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

const int INF=0x3f3f3f3f;
const int N=50005;
const int M=500005;
int top;
int dist[N],pre[N];
bool vis[N];
int c[N];
int maxflow;

struct Vertex{
    int first;
}V[N];
struct Edge{
    int v,next;
    int cap,flow,cost;
}E[M];

void init(){
    memset(V,-1,sizeof(V));
    top=0;
    maxflow=0;
}

void add_edge(int u,int v,int c,int cost){
    E[top].v=v;
    E[top].cap=c;
    E[top].flow=0;
    E[top].cost=cost;
    E[top].next=V[u].first;
    V[u].first=top++;
}

void add(int u,int v,int c,int cost){
    add_edge(u,v,c,cost);
    add_edge(v,u,0,-cost);
}

bool SPFA(int s,int t,int n){
    int i,u,v;
    queue<int>qu;
    memset(vis,false,sizeof(vis));
    memset(c,0,sizeof(c));
    memset(pre,-1,sizeof(pre));
    for(i=1;i<=n;i++){
        dist[i]=INF;
    }
    vis[s]=true;
    c[s]++;
    dist[s]=0;
    qu.push(s);
    while(!qu.empty()){
        u=qu.front();
        qu.pop();
        vis[u]=false;
        for(i=V[u].first;~i;i=E[i].next){
            v=E[i].v;
            if(E[i].cap>E[i].flow&&dist[v]>dist[u]+E[i].cost){
                dist[v]=dist[u]+E[i].cost;
                pre[v]=i;
                if(!vis[v]){
                    c[v]++;
                    qu.push(v);
                    vis[v]=true;
                    if(c[v]>n){
                        return false;
                    }
                }
            }
        }
    }
    if(dist[t]==INF){
        return false;
    }
    return true;
}

int MCMF(int s,int t,int n){
    int d;
    int i,mincost;
    mincost=0;
    while(SPFA(s,t,n)){
        d=INF;
        for(i=pre[t];~i;i=pre[E[i^1].v]){
            d=min(d,E[i].cap-E[i].flow);
        }
        maxflow+=d;
        for(i=pre[t];~i;i=pre[E[i^1].v]){
            E[i].flow+=d;
            E[i^1].flow-=d;
        }
        mincost+=dist[t]*d;
    }
    return mincost;
}

int aa[150];
int bb[150];
int ab[150][150];

int main(){
    int n,m;
    int v,u,w,c;
    int s,t;
    cin>>m>>n;
    init();
    s=0,t=n*m+1;
    for(int i=1;i<=m;i++){
        cin>>aa[i];
        add(s,i,aa[i],0);
    }
    for(int i=1;i<=n;i++){
        cin>>bb[i];
        add(m+i,t,bb[i],0);
    }
    for(int i=1;i<=m;i++){
        for(int j=1;j<=n;j++){
            cin>>ab[i][j];
            add(i,m+j,INF,ab[i][j]);
        }
    }
    int ans=MCMF(s,t,n*m+2);
    cout<<ans<<endl;
    init();
    for(int i=1;i<=m;i++){
        add(s,i,aa[i],0);
    }
    for(int i=1;i<=n;i++){
        add(m+i,t,bb[i],0);
    }
    for(int i=1;i<=m;i++){
        for(int j=1;j<=n;j++){
            add(i,m+j,INF,-ab[i][j]);
        }
    }
    ans=MCMF(s,t,n*m+2);
    cout<<-ans<<endl;
}