网络流——最大流 EK算法模板 P3376 【模板】网络最大流

复杂度最坏:O(n*m^2)  因为最多会找n*m次增广路

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int M = 5000+7;
const int N = 200+7;
int head[N],cnt=1;
void init(){cnt=1,memset(head,0,sizeof(head));}
struct EDGE{int to,nxt;ll w;}ee[M*2];
void add(int x,int y,ll w){ee[++cnt].nxt=head[x],ee[cnt].w=w,ee[cnt].to=y,head[x]=cnt;}
const int INF =1<<29;
int n,m,s,t;
ll maxflow;
int v[N],pre[N];
ll incf[N];
bool bfs()
{
	memset(v,0,sizeof(v));
	queue<int>q;
	q.push(s);v[s]=1;
	incf[s]=INF;
	while(q.size())
	{
		int x=q.front();q.pop();
		for(int i=head[x];i;i=ee[i].nxt)
		{
			int y=ee[i].to;
			ll w=ee[i].w;
			if(w)
			{
				if(v[y])continue;
				incf[y]=min(incf[x],w);//到点y的最大流量
				pre[y]=i;//记录前驱,我们只要找到任意一条增广路即可
				q.push(y),v[y]=1;
				if(y==t)return true; 
			}
		}
	}
	return false;
}
void update()
{
	int x=t;
	while(x!=s)
	{
		int i=pre[x];
		ee[i].w-=incf[t];
		ee[i^1].w+=incf[t];
		x=ee[i^1].to;
	}
	maxflow+=incf[t];
}
int mp[N][N];
int main()
{
	cin>>n>>m>>s>>t;
	memset(head,0,sizeof(head));
	cnt=1,maxflow=0;
	for(int i=1;i<=m;i++)
	{
		int x,y;ll w;
		scanf("%d%d%lld",&x,&y,&w);
		if(mp[x][y]==0)
		{
			add(x,y,w);mp[x][y]=cnt;
			add(y,x,0);
		}
		else ee[mp[x][y]].w+=w;
	}
	while(bfs())update();
	cout<<maxflow<<endl;
	return 0;
}

猜你喜欢

转载自blog.csdn.net/bjfu170203101/article/details/107304598