网络流Dinic算法模板(各种优化)

#include 
#include 
#include 
#include 
#define inf 0x3f3f3f3f3f3f3f3f
#define ll long long
#define MAXN 10005
using namespace std;
int n,m;//点数、边数 
int sp,tp;//原点、汇点 
struct node
{
	int v,next;
	ll cap;
}mp[MAXN*10];
int pre[MAXN],dis[MAXN],cur[MAXN];//cur为当前弧优化,dis存储分层图中每个点的层数(即到原点的最短距离),pre建邻接表
int cnt=0;
void init()//不要忘记初始化
{
	cnt=0;
	memset(pre,-1,sizeof(pre));
}
void add(int u,int v,int w)//加边 
{
	mp[cnt].v=v;
	mp[cnt].cap=w;
	mp[cnt].next=pre[u];
	pre[u]=cnt++;
}
bool bfs()//建分层图
{
	memset(dis,-1,sizeof(dis));
	queueq;
	while(!q.empty())
	q.pop();
	q.push(sp);
	dis[sp]=0;
	int u,v;
	while(!q.empty())
	{
		u=q.front();
		q.pop();
		for(int i=pre[u];i!=-1;i=mp[i].next)
		{
			v=mp[i].v;
			if(dis[v]==-1&&mp[i].cap>0)
			{
				dis[v]=dis[u]+1;
				q.push(v);
				if(v==tp)
				break;
			}
		}
	}
	return dis[tp]!=-1;
}
ll dfs(int u,ll cap)//寻找增广路
{
	if(u==tp||cap==0)
	return cap;
	ll res=0,f;
	for(int &i=cur[u];i!=-1;i=mp[i].next)
	{
		int v=mp[i].v;
		if(dis[v]==dis[u]+1&&(f=dfs(v,min(cap-res,mp[i].cap)))>0)
		{
			mp[i].cap-=f;
			mp[i^1].cap+=f;
			res+=f;
			if(res==cap)
			return cap;
		}
	}
	if(!res)
	dis[u]=-1;
	return res;
}
ll dinic()
{
	ll ans=0;
	while(bfs())
	{
		for(int i=1;i<=n;i++)
		cur[i]=pre[i];
		ans+=dfs(sp,inf);
	}
	return ans;
}
int main()
{
	
return 0;
}

你可能感兴趣的:(各种模板)