P3376 【模板】网络最大流

最大流EK板子

对板子的一个小优化。取edge的时候多加一个const.从洛谷的500ms吸氧气加到260ms

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
typedef long long LL;
const int maxn=110000;
const int INF=0x3f3f3f3f;
struct edge{
	int from,to,cap,flow;//起点，终点，容量，流量 
};
vector<edge>e;//保存所有边的信息 
vector<int>g[maxn];//邻接表，g[i][j]保存节点i的第j条边在e数组里的编号 
int a[maxn];//每个点目前流经的水量
int p[maxn];//p[i]表示从原点s到终点t的节点i的前一条边的编号 

//void init(int n){
//	for(int i=0;i<=n;i++) g[i].clear();
//	e.clear();
//}
void addedge(int u,int v,int c){
	e.push_back({u,v,c,0});//正向边
	e.push_back({v,u,0,0});//反向边，容量是0
	int m=e.size();
	g[u].push_back(m-2);//正向边编号，起始号0 
	g[v].push_back(m-1); //反向边编号，起始号1 
}
LL Maxflow(int s,int t){
	LL flow=0;
	while(true){
		memset(a,0,sizeof(a));//从原点s开始放水。最初每个点的水量为0
		queue<int>Q;
		Q.push(s);
		a[s]=INF; //原点的水设置成无穷大
		while(Q.size()){
			int x=Q.front();//取出目前水流到的节点
			Q.pop();
			for(int i=0;i<g[x].size();i++){//所有邻接节点 
				const edge& now=e[g[x][i]];//用const 快一点 
				if(!a[now.to]&&now.cap>now.flow){//a[i]=0表示还没流到，cap>flow说明路还没满 
					p[now.to]=g[x][i];//反向记录路径
					a[now.to]=min(a[x],now.cap-now.flow);//流到下一点的水量为上一点的水量或者路径上还可以流的最大流量，这两者取最小值
					Q.push(now.to); 
				}
			} 
			if(a[t]) break;//如果已经流到了终点t，退出本次找增广路 
		} 
		if(!a[t]) break;//如果所有路都已经试过，水不能流到终点 ，说明没有增广路，已经是最大流 
		for(int u=t;u!=s;u=e[p[u]].from)//反向记录路径
		{
			e[p[u]].flow+=a[t];//路径上所有正向边的流量增加流到终点的流量
			e[p[u]^1].flow-=a[t];//路径上所有反向边的流量减少流到终点的流量	
		} 
		flow+=a[t];//最大流加上本次流到终点的流量
	}
	return flow;	
}
int main(void)
{
 // cin.tie(0);std::ios::sync_with_stdio(false);
  int n,m,s,t;//n为点数,m为边数
  scanf("%d%d%d%d",&n,&m,&s,&t);
  for(int i=1;i<=m;i++){
  	int _x,_y,_w;scanf("%d%d%d",&_x,&_y,&_w);
  	addedge(_x,_y,_w);
  }
 // cout<<Maxflow(s,t)<<endl;
 printf("%lld\n",Maxflow(s,t));
return 0;
}