最大流当前弧优化Dinic模板

时间:2022-03-03 05:39:56

最大流模板:

  • 普通最大流
  • 无向图限制:将无向图的边拆成2条方向相反的有向边
  • 顶点有流量限制:拆成2个点,连接一条容量为点容量限制的边
  • 无源汇点有最小流限制的最大流:理解为水管流量形成循环
  • 有源汇点的最小流限制的最大流
  • 有源汇点的最小流限制的最小流
  • 容量为负数:不能直接利用最大流求边权为负数的最小割。不知道怎么具体处理。。。

模板使用Dinic分层算法,使用了当前弧优化,效率还是不错的,使用的是vector存图,如果使用邻接表存图效率应该会更高吧。

但是ispa算法的效率更高,采用了当前弧优化的ispa算法那就更更高的效率了(=^ ^=),但是这个算法目前不会啊。。。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<set>
#include<map>
#include<queue>
#include<stack>
#include<vector>
using namespace std;
typedef long long ll;
typedef pair<int,int> P;
#define PI acos(-1.0)
const int maxn=1e5+,maxm=1e5+,inf=0x3f3f3f3f,mod=1e9+;
const ll INF=1e18+;
struct edge
{
int from,to,cap,flow;
};
vector<edge>es;
vector<int>G[maxn];
bool vis[maxn];
int dist[maxn];
int iter[maxn];
void init(int n)
{
for(int i=; i<=n+; i++) G[i].clear();
es.clear();
}
void addedge(int from,int to,int cap)
{
es.push_back((edge)
{
from,to,cap,
});
es.push_back((edge)
{
to,from,,
});
int x=es.size();
G[from].push_back(x-);
G[to].push_back(x-);
}
bool BFS(int s,int t)
{
memset(vis,,sizeof(vis));
queue <int> Q;
vis[s]=;
dist[s]=;
Q.push(s);
while(!Q.empty())
{
int u=Q.front();
Q.pop();
for (int i=; i<G[u].size(); i++)
{
edge &e=es[G[u][i]];
if (!vis[e.to]&&e.cap>e.flow)
{
vis[e.to]=;
dist[e.to]=dist[u]+;
Q.push(e.to);
}
}
}
return vis[t];
}
int DFS(int u,int t,int f)
{
if(u==t||f==) return f;
int flow=,d;
for(int &i=iter[u]; i<G[u].size(); i++)
{
edge &e=es[G[u][i]];
if(dist[u]+==dist[e.to]&&(d=DFS(e.to,t,min(f,e.cap-e.flow)))>)
{
e.flow+=d;
es[G[u][i]^].flow-=d;
flow+=d;
f-=d;
if (f==) break;
}
}
return flow;
}
int Maxflow(int s,int t)
{
int flow=;
while(BFS(s,t))
{
memset(iter,,sizeof(iter));
int d=;
while(d=DFS(s,t,inf)) flow+=d;
}
return flow;
} int in[maxn],out[maxn];
int l[maxn];
int main()
{
int n,m;
scanf("%d%d",&n,&m); /**有源汇点*/
int s,t;
scanf("%d%d",&s,&t); memset(in,,sizeof(in));
memset(out,,sizeof(out));
init(n);
for (int i=; i<=m; i++)
{
int u,v,r;
scanf("%d %d %d %d",&u,&v,&l[i],&r);
out[u]+=l[i],in[v]+=l[i];
addedge(u,v,r-l[i]);
} /**有源汇点有下界*/
addedge(t,s,inf); /**有下界添加新源点0,汇点n+1*/
int ss=,tt=n+;
int sum=;
for(int i=; i<=n; i++)
{
if(in[i]>out[i])
{
addedge(ss,i,in[i]-out[i]);
sum+=in[i]-out[i];
}
else if(in[i]<out[i])
addedge(i,tt,out[i]-in[i]);
} /**无下界直接用s,t求最大流*/
if(Maxflow(ss,tt)!=sum) puts("NO");
else
{
puts("YES"); /**有源汇点*/
int ans=-es[*m+].flow;
es[*m+].flow=;
es[*m].flow=;
es[*m].cap=;
ans+=Maxflow(s,t);///ans-=Maxflow(t,s);///最小流
printf("%d\n",ans); /**无源汇点:可以理解成水管形成循环*/
for(int i=; i<m*; i+=)
printf("%d\n",es[i].flow+l[i/+]);///每根水管的流量情况
}
return ;
}