#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=5005;
const int inf=0x3f3f3f3f;
int tot,n,m,s,t,x,y,z,d[maxn];
int adj[maxn],nex[maxn<<1],to[maxn<<1],w[maxn<<1],flow[maxn<<1];
void add(int x,int y,int z){
nex[tot]=adj[x];
adj[x]=tot;
to[tot]=y;
w[tot]=z;
flow[tot++]=0;
}
bool bfs(){
memset(d,0,sizeof(d));
queue<int> q;
d[s]=1;
q.push(s);
while(!q.empty()){
int u=q.front();
q.pop();
for(int i=adj[u];~i;i=nex[i]){
int v=to[i];
if(!d[v]&&w[i]>flow[i]){
d[v]=d[u]+1;
q.push(v);
if(v==t) return 1;
}
}
}
return 0;
}
int dfs(int u,int fw){
if(u==t) return fw;
int rest=fw;
for(int i=adj[u];~i&&rest;i=nex[i]){
int v=to[i];
if(d[v]==d[u]+1&&w[i]>flow[i]){
int k=dfs(v,min(rest,w[i]-flow[i]));
if(!k) d[v]=0;
flow[i]+=k;
flow[i^1]-=k;
rest-=k;
}
}
return fw-rest;
}
int dinic(){
int maxflow=0;
while(bfs()){
//for(int i=1;i<=n;i++)
maxflow+=dfs(s,inf);
}
return maxflow;
}
signed main()
{
cin>>n>>m>>s>>t;
memset(adj,-1,sizeof(adj));
for(int i=1;i<=m;i++){
cin>>x>>y>>z;
add(x,y,z);add(y,x,0);
}
cout<<dinic();
return 0;
}
注意几个细节:adj初始化为-1,tot下标从0开始;~i
加上当前弧优化
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=5005;
const int inf=0x3f3f3f3f;
int tot,n,m,s,t,x,y,z,d[maxn],now[maxn];
int adj[maxn],nex[maxn<<1],to[maxn<<1],w[maxn<<1],flow[maxn<<1];
void add(int x,int y,int z){
nex[tot]=adj[x];
adj[x]=tot;
to[tot]=y;
w[tot]=z;
flow[tot++]=0;
}
bool bfs(){
memset(d,0,sizeof(d));
queue<int> q;
d[s]=1;
q.push(s);
for(int i=1;i<=n;i++)
now[i]=adj[i];
while(!q.empty()){
int u=q.front();
q.pop();
for(int i=adj[u];~i;i=nex[i]){
int v=to[i];
if(!d[v]&&w[i]>flow[i]){
d[v]=d[u]+1;
q.push(v);
if(v==t) return 1;
}
}
}
return 0;
}
int dfs(int u,int fw){
if(u==t) return fw;
int rest=fw;
for(int i=now[u];~i&&rest;i=nex[i]){
int v=to[i];
if(d[v]==d[u]+1&&w[i]>flow[i]){
int k=dfs(v,min(rest,w[i]-flow[i]));
if(!k) d[v]=0;
flow[i]+=k;
flow[i^1]-=k;
rest-=k;
}
}
return fw-rest;
}
int dinic(){
int maxflow=0;
while(bfs()){
//for(int i=1;i<=n;i++)
maxflow+=dfs(s,inf);
}
return maxflow;
}
signed main()
{
cin>>n>>m>>s>>t;
memset(adj,-1,sizeof(adj));
for(int i=1;i<=m;i++){
cin>>x>>y>>z;
add(x,y,z);add(y,x,0);
}
cout<<dinic();
return 0;
}