EK求最大流

最爱丁珰發表於2024-09-01

程式碼見下

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=1010,M=10010,inf=1<<29;
int End[M<<1],Next[M<<1],Len[M<<1],Last[N];
bool v[N];
int n,m,s,t,cnt=1,incf[N],pre[N];//注意cnt從1開始 
ll maxflow=0;
queue<int> q;
void add(int x,int y,int z)//最開始的流全為0,所以正向邊為z反向邊為0 
{
	End[++cnt]=y,Next[cnt]=Last[x],Len[cnt]=z,Last[x]=cnt;
	End[++cnt]=x,Next[cnt]=Last[y],Len[cnt]=0,Last[y]=cnt;
}
bool bfs()
{
	while(!q.empty()) q.pop();
	memset(v,0,sizeof(v));
	q.push(s),v[s]=1;
	incf[s]=inf;//incf[x]表示當前的殘存網路上從s到x的路徑的最小邊長度 
	while(!q.empty())
	{
		int x=q.front();
		q.pop();
		for(int i=Last[x];i;i=Next[i])
		{
			int y=End[i],l=Len[i];
			if(!Len[i]||v[y]) continue;
			//若Len[i]為0表示這條邊不存在於殘存網路中,所以直接跳過 
			incf[y]=min(incf[x],l);
			pre[y]=i;//記錄當前殘存網路中y的前驅,之後好更新 
			q.push(y),v[y]=1;
			if(y==t) return 1;
		}
	}
	return 0;
}
void update()
{
	int x=t;
	while(x!=s)
	{
		int i=pre[x];
		Len[i]-=incf[t],Len[i^1]+=incf[t];//更新殘存網路 
		x=End[i^1];
	}
	maxflow+=incf[t];
}
int main()
{
	scanf("%d%d%d%d",&n,&m,&s,&t);
	for(int i=1;i<=m;i++)
	{
		int u,v,c;
		scanf("%d%d%d",&u,&v,&c);
		add(u,v,c);
	}
	while(bfs()) update();
	printf("%lld",maxflow);
    return 0;
}

相關文章