最大流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;
}