[Luogu] 运输问题 -- 00

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

#include <bits/stdc++.h>

#define gc getchar()

using namespace std;
const int N = ;
const int oo = ;

int n, m, S, T, now;
int head[N], dis[N], C[N][N], A[N], B[N], pre[N];
bool vis[N * N];
struct Node{int u, v, cost, flow, nxt;} G[(N * N) << ], E[(N * N) << ];
queue <int> Q;

inline int read(){
    int x = ; char c = gc;
    while(c < '' || c > '') c = gc;
    while(c >= '' && c <= '') x = x *  + c - '', c = gc;
    return x;
}

inline void add(int u, int v, int flow, int cost){
    G[now].u = u; G[now].v = v; G[now].flow = flow; G[now].cost = cost;
    G[now].nxt = head[u]; head[u] = now ++;
}

inline void build_G(){
    for(int i = ; i <= n; i ++) add(S, i, oo, ), add(i, S, , );
    for(int i = ; i <= n; i ++)
        for(int j = ; j <= m; j ++)
            add(i, n + j, A[i], C[i][j]), add(n + j, i, , - C[i][j]);
    for(int i = ; i <= m; i ++) add(n + i, T, B[i], ), add(T, n + i, , );
    for(int i = ; i <= now; i ++){
        E[i].u = G[i].u; E[i].v = G[i].v; E[i].nxt = G[i].nxt;
        E[i].cost = - G[i].cost; E[i].flow = G[i].flow;
    }
}

inline bool spfa(int start, int endd){
    for(int i = S; i <= T; i ++) vis[i] = , dis[i] = oo;
    dis[start] = ;
    Q.push(start);
    while(!Q.empty()){
        int topp = Q.front();
        Q.pop(); vis[topp] = ;
        for(int i = head[topp]; ~ i; i = G[i].nxt){
            int v = G[i].v;
            if(dis[v] > dis[topp] + G[i].cost && G[i].flow > ){
                dis[v] = dis[topp] + G[i].cost;
                pre[v] = i;
                if(!vis[v]) vis[v] = , Q.push(v);
            }
        }
    }
    return dis[endd] != oo;
}

inline void Mfmc(int & Cost){
    while(spfa(S, T)){
        int endd = T, Now;
        int min_f = oo;
        while() {
            Now = pre[endd];
            min_f = min(min_f, G[Now].flow);
            if(G[Now].u == S) break;
            endd = G[Now].u;
        }
        Cost += dis[T] * min_f;
        endd = T;
        while(){
            Now = pre[endd];
            G[Now].flow -= min_f;
            G[Now ^ ].flow += min_f;
            if(G[Now].u == S) break;
            endd = G[Now].u;
        }
    }
}

int main()
{
    n = read(); m = read(); T = n + m + ;
    for(int i = S; i <= T; i ++) head[i] = -;
    for(int i = ; i <= n; i ++) A[i] = read();
    for(int i = ; i <= m; i ++) B[i] = read();
    for(int i = ; i <= n; i ++) for(int j = ; j <= m; j ++) C[i][j] = read();
    build_G();
    int Min_cost();
    Mfmc(Min_cost);
    cout << Min_cost << endl;
    for(int i = ; i <= now; i ++){
        G[i].u = E[i].u; G[i].v = E[i].v; G[i].nxt = E[i].nxt;
        G[i].cost = E[i].cost; G[i].flow = E[i].flow;
    }
    Min_cost = ;
    Mfmc(Min_cost);
    cout << - Min_cost;
    return ;
}
/*
2 3
220 280
170 120 210
77 39 105
150 186 122
*/