网络流之当前弧优化浅谈

Posted 2020-11-15 kamimxr

在dinic中，我们会发现，dfs中每条边都会遍历至少一遍，那么我们可以把一定不会用到的边删去吗？

答案是当然可以，这就用到了当前弧优化；

其实这个优化在很久很久以前学习欧拉回路的时候就接触到了；

每次增广一条路后可以看做“榨干”了这条路，既然榨干了就没有再增广的可能了。但如果每次都扫描这些“枯萎的”边是很浪费时间的。那我们就记录一下“榨取”到那条边了，然后下一次直接从这条边开始增广，就可以节省大量的时间。这就是 当前弧优化 。

实现方法就是每次dfs时将head数组复制一遍，然后再新数组中跑dfs

#include <bits/stdc++.h>
#define int long long
using namespace std;
struct littlestar{
	int to;
	int nxt;
	int w;
}star[10010];
int head[10010],cur[10010],cnt=1;
void add(int u,int v,int w)
{
	star[++cnt].to=v;
	star[cnt].nxt=head[u];
	star[cnt].w=w;
	head[u]=cnt;
}
int s,t;
queue<int> q;
int n;
int dep[10010];
bool bfs()
{
	for(int i=1;i<=n;i++){
		dep[i]=0;
		cur[i]=head[i];
	}
	while(q.size()) q.pop();
	q.push(s);
	dep[s]=1;
	while(q.size()){
		int u=q.front();
		q.pop();
		for(int i=head[u];i;i=star[i].nxt){
			int v=star[i].to;
			if(!dep[v]&&star[i].w){
				dep[v]=dep[u]+1;
				q.push(v);
				if(v==t) return 1;
			}
		}
	}
	return 0;
}
int m;
int dinic(int u,int flow)
{
	if(u==t) return flow;
	int rest=flow,tmp;
	for(int i=cur[u];i&&rest;i=star[i].nxt){
		cur[u]=i;
		int v=star[i].to;
		if(dep[v]==dep[u]+1&&star[i].w){
			tmp=dinic(v,min(star[i].w,rest));
			if(!tmp) dep[v]=0;
			star[i].w-=tmp;
			star[i^1].w+=tmp;
			rest-=tmp;
		}
	}
	return flow-rest;
}
signed main()
{
	cin>>n>>m>>s>>t;
	for(register int i=1;i<=m;i++){
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		add(u,v,w);
		add(v,u,0);
	}
	int maxflow=0;
	while(bfs()){
		int flow;
		while(flow=dinic(s,LLONG_MAX)) maxflow+=flow;
	}
	cout<<maxflow;
}