【code】最小费用最大流(Dinic / zkw) 模板
我在网上找到的大部分标题为Dinic的费用流感觉都像是EK的费用流,而一些真的Dinic费用流模板中的变量、函数命名又太冗长,不能很直观地理解。因此我自己写了一个基于Dinic最大流(后来查了是zkw)的费用流模板,实测比EK费用流快30%-40%,希望有帮助。
#include <bits/stdc++.h>
using namespace std;
const int INF = 0x3f3f3f3f;
// edge
struct Edge{
int from, to, f, w;
}E[1000005];
int Hed[100005], Nex[1000005], ct=1, Cur[100005];
void Add(int a, int b, int f, int w){ //加边
E[++ct].from=a, E[ct].to=b, E[ct].f=f, E[ct].w=w, Nex[ct]=Hed[a], Hed[a]=ct;
E[++ct].from=b, E[ct].to=a, E[ct].f=0, E[ct].w=-w, Nex[ct]=Hed[b], Hed[b]=ct;
}
// mincostmaxflow
int n, m, s, t, maxflow, mincost, Dis[100005], F[100005];
bool SPFA(){ //最短路分层,从汇点向源点分层能保证DFS走的一定是最短路,不会浪费时间走错路
queue<int> Q; Q.push(s);
memset(Dis, INF, sizeof Dis);
Dis[s] = 0; int k;
while(!Q.empty()){
k = Q.front(); Q.pop();
F[k] = 0;
for(int i=Hed[k]; i; i=Nex[i]){
if(E[i].f && Dis[k]+E[i].w<Dis[E[i].to]){
Dis[E[i].to] = Dis[k]+E[i].w;
if(!F[E[i].to])
Q.push(E[i].to), F[E[i].to] = 1;
}
}
}
return Dis[t] != INF;
}
int DFS(int k, int flow){
if(k == t){maxflow += flow; return flow;} //达到汇点更新最大流
int sum = 0; F[k] = 1; //F[]保证了当出现 0 费用边的时候不会出现两个点之间来回跑的情况
for(int i=Cur[k]; i; i=Nex[i]){
if(!F[E[i].to] && E[i].f && Dis[E[i].to]==Dis[k]+E[i].w){
Cur[k] = i; //当前弧优化
int p = DFS(E[i].to, min(flow-sum, E[i].f));
sum += p, E[i].f -= p, E[i^1].f += p, mincost += p*E[i].w; //更新费用
if(sum == flow) break;
}
}
F[k] = 0;
return sum;
}
void Dinic(){
while(SPFA()){
memcpy(Cur, Hed, sizeof Hed);
DFS(s, INF);
}
}
// main
int main(){
int a, b, c, d;
scanf("%d%d%d%d", &n, &m, &s, &t);
for(int i=1; i<=m; ++i){
scanf("%d%d%d%d", &a, &b, &c, &d);
Add(a, b, c, d);
}
Dinic();
printf("%d %d", maxflow, mincost);
return 0;
}