洛谷P3376 （最大流模板）

#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;
}

 

---

如果您觉得阅读本文对您有帮助，请点一下“推荐”按钮，您的“推荐”将是我最大的写作动力！欢迎各位转载，但是未经作者本人同意，转载文章之后必须在文章页面明显位置给出作者和原文连接，否则保留追究法律责任的权利。