自己整理,仅做模板保存。
cur【】数组是当前弧优化的辅助数组。
当前弧优化是dinic算法中十分重要的优化。cur【】数组的使用方法即首先将head【】数组复制一遍。在dfs中再改变当前榨取到哪一条边。每次榨取,该边所有流量必定被榨干,下一次搜索,直接从这条边开始就好。
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=105;
const int maxe=10010;
const ll inf=1e15;
int dep[maxn];
int head[maxn];
int cur[maxn]; //当前弧优化的辅助数组
int step;
struct ED
{
int to,next;
ll val;
}edge[maxe];
void add(int t,ll v,int x)
{
edge[step].to=t;
edge[step].val=v;
edge[step].next=head[x];
head[x]=step;
step++;
}
bool bfs(int s,int t)
{
memset(dep,0,sizeof(dep));
queue<int>Q;
Q.push(s);
dep[s]=1;
cur[s]=head[s];
while(!Q.empty())
{
s=Q.front();
Q.pop();
for(int i=head[s];i!=-1;i=edge[i].next)
{
int y=edge[i].to;
if(edge[i].val>0&&dep[y]==0)
{
dep[y]=dep[s]+1;
cur[y]=head[y];
if(y==t) return true;
Q.push(y);
}
}
}
return false;
}
ll dfs(int s,ll flow,int t)
{
if(s==t||flow<=0) return flow;
ll rest=flow;
for(int i=cur[s];i!=-1;i=edge[i].next)
{
cur[s]=i; //弧优化记录当前“榨取”到哪条边了
int y=edge[i].to;
if(edge[i].val>0&&dep[y]==dep[s]+1)
{
ll temp=dfs(y,min(rest,edge[i].val),t);
if(temp<=0) dep[y]=0; //该点无法再提供流量,删掉
rest-=temp;
edge[i].val-=temp;
edge[i^1].val+=temp;
if(rest<=0) break;
}
}
return flow-rest;
}
ll dinic(int s,int t)
{
ll ans=0;
while(bfs(s,t))
ans+=dfs(s,inf,t);
return ans;
}
int main()
{
ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
int n,m,s,t;
cin>>n>>m>>s>>t;
step=2; //从偶数开始记录边,方便记录反向边
memset(head,-1,sizeof(head));
for(int i=1;i<=m;i++)
{
int x,y;
ll f;
cin>>x>>y>>f;
add(y,f,x);
add(x,0,y);
}
ll ans=dinic(s,t);
cout<<ans<<endl;
return 0;
}