2023CCPC女生赛 D - 金人旧巷市廛喧

最大费用可行流。

官方题解已经很详尽了，主要来贴个代码。

solution

#include <bits/stdc++.h>
using namespace std;
#define debug cout<<"!!!"<<endl;
#define rep(i, a, b) for (int i = (a); i <= (b); i++)
#define FOR(i, a, b) for (int i = (a); i <= (b); i++)
#define ROF(i, a, b) for (int i = (a); i >= (b); i--)
#define ll long long
const int inf = 1<<30;
const int N = 2001,M = 50;
int n,m,k,a[M][M],idx1[M][M],idx2[M][M];
ll val[N][N],cost[N][N],s,t, mincost=0;
ll pre[N],flow[N],dis[N],vis[N];
vector<int> g[N];
int dx[4] = {0,1,0,-1},dy[4] = {1,0,-1,0}; 

 
int spfa(){
    FOR(i,1,n) dis[i] = inf; //离s点距离，初始是inf
    FOR(i,1,n) pre[i] = 0; //记录前驱，0表示一开始没有前驱
    dis[s] = 0; flow[s] = inf; vis[s] = 1; //超级原点的初始设置
    queue<int> q; q.push(s); //队列中放入原点
    while(q.size()){
        int u = q.front(); q.pop();
        vis[u] = 0; //出队，vis标记为0
        for(int v:g[u]){
            if(val[u][v] <= 0) continue; //流量<=0，实际上不连通
            if(dis[v] <= dis[u]+cost[u][v]) continue; //松弛失败，不做任何操作
            dis[v] = dis[u]+cost[u][v]; //进行松弛，更新距离
            pre[v] = u; //记录前驱（为了服务于mcmf函数，要记录路径）
            flow[v] = min(flow[u], val[u][v]); //记录可行的最大流量（val的前缀最小）
            if(!vis[v]) q.push(v), vis[v]=1; //当前点进入队列
        }
    }
    if(dis[t] == inf || dis[t]>=0) return 0; //如果无法到终点，或者到终点不赚钱或亏钱了，那以后就都不继续走了（证明略）
    //重点！！！dis[t]>=0就是【最小费用可行流】比起【最小费用最大流】需要加的唯一一句代码！！！
    else return 1; //能到终点，还能赚钱，这次走比不走好
}
void mcmf(){
    while(spfa()){
        int now = t;
        //maxflow += flow[t]; 记录最大流（这里其实跑的不是最大流，这个值没有意义，仅作为板子参考）
        mincost += flow[t]*dis[t]; //flow是本次流量，dis是这一条路上的【单位流量的费用】的总和
        while(now!=s){ //从终点一路往前，更新流量的信息
            val[pre[now]][now] -= flow[t]; //正向路，流量减少
            val[now][pre[now]] += flow[t]; //反向路，流量增加
            now = pre[now]; //往前走
        }
    }
}

int get(int x,int y,int res){
    x--,y--;
    if(res == 1 )return (x*m+y+1);
    else return n*m+x*m+y+1;
}


signed main(){
    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    cin>>n>>m>>k; 

    rep(i,1,n)
        rep(j,1,m)
            cin >> a[i][j];

    rep(i,1,k){
        int x,y;cin >> x >> y;
        idx1[x][y] = 1;
    }
    rep(i,1,k){
        int p,q;cin >> p >> q;
        idx2[p][q] = 1;
    }
    // rep(i,1,n){
    //     rep(j,1,m){
    //         cout << get(i,j,1) << " ";
    //     }
    //     cout << endl;
    // }
    // rep(i,1,n){
    //     rep(j,1,m){
    //         cout << get(i,j,2) << " ";
    //     }
    //     cout << endl;
    // }

    rep(i,1,n){
        rep(j,1,m){

            
            int u = get(i,j,1),v = get(i,j,2);

            g[u].push_back(v);
            g[v].push_back(u);
            val[u][v] = 1; //注意只有单向流量
            if(a[i][j] == 1){
                    cost[u][v] = 0;
                    cost[v][u] = 0;
            }
            else{
                    cost[u][v] = 1;
                    cost[v][u] = -1;
            }

            rep(z,0,3){
                int xx = i+dx[z],yy = j + dy[z];
                if(xx < 1 || xx > n || yy < 1 || yy > m)continue;
                if(a[xx][yy] == -1)continue;

                int u1 = get(xx,yy,1);
                g[v].push_back(u1);
                g[u1].push_back(v);

                val[v][u1] = 1;
                cost[v][u1] = 0;
                cost[u1][v] = 0;
            }

        }
    }

    s=0; t=2*n*m+1; //超级原点和终点
    rep(i,1,n){
        rep(j,1,m){
            if(idx1[i][j]){
                int u = get(i,j,1);
                g[s].push_back(u); 
                 g[u].push_back(s); 

                val[s][u] = 1; 
                cost[s][u] = 0; 
                cost[u][s] = 0;
            }
        }
    }
    rep(i,1,n){
        rep(j,1,m){
            if(idx2[i][j]){
                int u = get(i,j,2);
                g[u].push_back(t);
                g[t].push_back(u); 

                val[u][t] = 1; 
                cost[u][t] = -100; 
                cost[t][u] = 100;
            }
        }
    }

    n = t;
    mcmf(); //跑mcmf
    cout<<-1*mincost<< endl; //输出最大费用可行流大小，注意乘上-1（因为代码里跑的是最小费用可行流）
}