EK算法——【模板】网络最大流

【模板】网络最大流

题目描述

如题，给出一个网络图，以及其源点和汇点，求出其网络最大流。

输入格式

第一行包含四个正整数 $n,m,s,t$，分别表示点的个数、有向边的个数、源点序号、汇点序号。

接下来 $m$ 行每行包含三个正整数 $u_i,v_i,w_i$，表示第 $i$ 条有向边从 $u_i$ 出发，到达 $v_i$，边权为 $w_i$（即该边最大流量为 $w_i$）。

输出格式

一行，包含一个正整数，即为该网络的最大流。

样例 #1

样例输入 #1

4 5 4 3
4 2 30
4 3 20
2 3 20
2 1 30
1 3 30

样例输出 #1

50

提示

样例输入输出 1 解释

题目中存在 $3$ 条路径：

• $4\to 2\to 3$，该路线可通过 $20$ 的流量。
• $4\to 3$，可通过 $20$ 的流量。
• $4\to 2\to 1\to 3$，可通过 $10$ 的流量（边 $4\to 2$ 之前已经耗费了 $20$ 的流量）。

故流量总计 $20+20+10=50$。输出 $50$。

---

数据规模与约定

• 对于 $30\%$ 的数据，保证 $n\le 10$，$m\le 25$。
• 对于 $100\%$ 的数据，保证 $1\le n\le 200$，$1\le m\le 5000$，$0\le w<2^{31}$。

代码及细节

//EK算法O(nm^2)
//最大流存在=|f|的条件为：
//不存在增广路
//增广路指的是从起点沿着容量>0的方向走（前提是残留网络）
//可行流不存在方向边
//残留网络是建立在可行流的基础上（可行流是没有增广路的），（此时存在方向边）
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
#define int long long
using namespace std;

const int N = 210,M = 5010*2,INT = 2e9;

int h[N],e[M],f[M],idx,ne[M];
bool st[N];
int pre[N];
int d[N];//表示每个节点到源点的最短费用
int n,m,S,T;

void add(int a,int b,int c){
    e[idx]=b,f[idx]=c,ne[idx]=h[a],h[a]=idx++;
    e[idx]=a,f[idx]=0,ne[idx]=h[b],h[b]=idx++;
}

//判断增广路，并记录此路径
bool bfs(){
    queue<int>q;
    
    memset(st,0,sizeof st);
    
    st[S]=true;
    
    d[S]=INT;
    
    q.push(S);
    
    while(q.size()){
        int t=q.front();
        q.pop();
        
        for(int i=h[t];~i;i=ne[i]){
            int ver=e[i];
            if(!st[ver]&&f[i]){//必须大于0
                st[ver]=true;
                d[ver]=min(d[t],f[i]);
                pre[ver]=i;
                if(ver==T)return true;
                q.push(ver);
            }
        }
    }
    return false;
}

//类似挤牙膏
//你先预处理出从起点到终点的，然后最后计算的时候从终点到起点
//为什么要取最小的，因为要花费最小
int EK(){
    
    int r=0;
    while(bfs()){
        r+=d[T];//累加最大流量，因为题目要求的是费用
        //用边的编号来推出点的编号
        for(int i=T;i!=S;i=e[pre[i]^1]){
            //为什么要这么减呢，因为我们要找到最大流/最大费用（自认为是容量最小的（每个值的容量可能会发生变化））
            f[pre[i]]-=d[T];
            f[pre[i]^1]+=d[T];
        }
    }
    return r;
}

signed main(){
    cin>>n>>m>>S>>T;
    
    memset(h,-1,sizeof h);
    
    for(int i=1;i<=m;i++){
        int a,b,c;
        cin>>a>>b>>c;
        add(a,b,c);
    }
    
    cout<<EK();
    
    return 0;
}