[网络流24题] 运输问题

Posted 2021-02-04 fengxunling

题目链接：戳我
最小费用最大流。
代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#define S 0
#define T n+m+1
#define MAXN 210
using namespace std;
int n,m,t=1,c,f;
int head[100010],pre_v[MAXN],pre_e[MAXN],dis[MAXN],done[MAXN];
int a[MAXN],b[MAXN],trans[MAXN][MAXN];
struct Edge{int nxt,to,dis,cost;}edge[100010];
inline void add(int from,int to,int dis,int cost)
{
    edge[++t].nxt=head[from],edge[t].to=to,edge[t].dis=dis,edge[t].cost=cost,head[from]=t;
    edge[++t].nxt=head[to],edge[t].to=from,edge[t].dis=0,edge[t].cost=-cost,head[to]=t;
}
inline bool spfa()
{
    memset(dis,0x3f,sizeof(dis));
    queue<int>q;
    q.push(S); dis[S]=0; done[S]=1; 
    while(!q.empty())
    {
        int u=q.front();q.pop();done[u]=0;
        for(int i=head[u];i;i=edge[i].nxt)
        {
            int v=edge[i].to;
            if(edge[i].dis&&dis[u]+edge[i].cost<dis[v])
            {
                dis[v]=dis[u]+edge[i].cost;
                pre_v[v]=u,pre_e[v]=i;
                if(!done[v]) q.push(v),done[v]=1;
            }
        }
    }
    if(dis[T]==0x3f3f3f3f) return false;
    int flow=(int)1e9;
    for(int i=T;i!=S;i=pre_v[i]) flow=min(flow,edge[pre_e[i]].dis);
    for(int i=T;i!=S;i=pre_v[i]) edge[pre_e[i]].dis-=flow,edge[pre_e[i]^1].dis+=flow;
    c+=flow*dis[T];
    f+=flow;
    return true;
}
int main()
{
    #ifndef ONLINE_JUDGE
    freopen("ce.in","r",stdin);
    #endif
    scanf("%d%d",&m,&n);
    for(int i=1;i<=m;i++) scanf("%d",&a[i]);
    for(int i=1;i<=n;i++) scanf("%d",&b[i]);
    for(int i=1;i<=m;i++)
        for(int j=1;j<=n;j++)
            scanf("%d",&trans[i][j]);
    for(int i=1;i<=m;i++) add(S,i,a[i],0);
    for(int i=1;i<=n;i++) add(i+m,T,b[i],0);
    for(int i=1;i<=m;i++)
        for(int j=1;j<=n;j++)
            add(i,j+m,(int)1e9,trans[i][j]);
    while(spfa());
    printf("%d
",c);
    memset(head,0,sizeof(head));
    t=1,c=0,f=0;
     for(int i=1;i<=m;i++) add(S,i,a[i],0);
    for(int i=1;i<=n;i++) add(i+m,T,b[i],0);
    for(int i=1;i<=m;i++)
        for(int j=1;j<=n;j++)
            add(i,j+m,(int)1e9,-trans[i][j]);
    while(spfa());
    printf("%d
",-c);
    return 0;
}