Dinic模板 最大流最小割 网络流 两个版本 加当前弧优化

一：bilibili关注up主：蒟蒻算法小课堂的版本

#include <bits/stdc++.h>

using namespace std;
const int maxn=220,maxm=440,inf=0x7f7f7f7f;

struct edge
{
    int to,w,next;
}e[maxm];

int n,m,cnt,head[maxn],dis[maxn],q[maxn];

void init()
{
    cnt=0;
    memset(head,-1,sizeof(head));
}

void adde(int u,int v,int w)
{
    e[cnt].to=v;e[cnt].w=w;e[cnt].next=head[u];head[u]=cnt++;
    e[cnt].to=u;e[cnt].w=0;e[cnt].next=head[v];head[v]=cnt++;
}

int bfs()
{
    memset(dis,-1,sizeof(dis));
    int front=0,tail=0;
    dis[1]=0;
    q[tail++]=1;
    while(front<tail)
    {
        int u=q[front++];
        if(u==n)return 1;
        //~用于计算时， 是按位取反运算符,比如 ~10101111 结果是01010000
        for(int i=head[u];~i;i=e[i].next)
        {
            int& v=e[i].to,w=e[i].w;
            if(dis[v]==-1&&w)
            {
                dis[v]=dis[u]+1;
                q[tail++]=v;
            }
        }
    }
    return 0;
}

int dfs(int s,int t,int flow)
{
    if(s==t)return flow;
    int pre=0;
    for(int i=head[s];~i;i=e[i].next)
    {
        int& v=e[i].to,w=e[i].w;
        if(dis[s]+1==dis[v]&&w>0)
        {
            int tmp=min(flow-pre,w);
            int tf=dfs(v,t,tmp);
            e[i].w-=tf;
            e[i^1].w+=tf;
            pre+=tf;
            if(pre==flow)return pre;
        }
    }
    return pre;
}

int dinic()
{
    int ret=0;
    while(bfs())ret+=dfs(1,n,inf);
    return ret;
}

int main()
{
    ios::sync_with_stdio(0);
    int u,v,w;
    scanf("%d %d",&m,&n);
    init();
    for(int i=0;i<m;i++)
    {
        scanf("%d %d %d",&u,&v,&w);
        adde(u,v,w);
    }
    printf("%d\n",dinic());
    return 0;
}

/*
5 4
1 2 40
1 4 20
2 4 20
2 3 30
3 4 10
50
*/

二：自己一开始学网络流，总结的模板，后来在洛谷做题被卡，再次参考别的博客的代码，改写dfs()函数，并且加入当前弧优化

#include <bits/stdc++.h>
#define xuhao(i,j) ((i-1)*m+j)
//Dinic+当前弧优化

using namespace std;
const int maxn=4e4+9e3+6e2+2,maxm=2e5+7e4+7e3+6e2+1,inf=0x7fffffff;
int n,m,a,s,t,tot=1,head[maxn],cur[maxn],dep[maxn],ans,sum,cnt;//用上了分层图，可以用dep判重了

struct edge
{
    int to,next,w;
}e[maxm];

void addedge(int x,int y,int w)
{
    e[++tot].to=y;e[tot].w=w;e[tot].next=head[x];head[x]=tot;
    e[++tot].to=x;e[tot].w=0;e[tot].next=head[y];head[y]=tot;
}

bool bfs()//bool 函数是一个小优化，判断是否能搜到汇点，如果连汇点都搜不到还dfs干什么？
{
    memset(dep,0,sizeof dep);//一定要初始化
    memcpy(cur,head,sizeof(head));
    queue<int>q;
    q.push(s);dep[s]=1;
    while(!q.empty())
    {
        int x=q.front();q.pop();
        for(int i=head[x];i;i=e[i].next)
        {
            int y=e[i].to,w=e[i].w;
            if(w&&!dep[y])//如果有残余流量（没有的话谁也过不去）并且这个点是第一次到达
            {
                dep[y]=dep[x]+1;
                q.push(y);
            }
        }
    }
    return dep[t];//t 的深度不为0，就是搜到了汇点
}

//int dfs(int x,int flow)
//{
//    if(x==t)return flow;
//    int sum=0;
//    for(int i=cur[x];i;i=e[i].next)
//    {
//        cur[x]=i;
//        int y=e[i].to,w=e[i].w;
//        if(w&&dep[y]==dep[x]+1)//仅允许流向下一层
//        {
//            int t=dfs(y,min(w,flow));
//            e[i].w-=t;e[i^1].w+=t;
//            flow-=t;sum+=t;
//        }
//    }
//    if(!sum)dep[x]=0;//我与终点（顺着残余网络）不连通的话，那么上一层的点请别给我流量
//    return sum;
//}

int dfs(int u,int flow) {
    if(u==t) return flow;
    int ans=0;
    for(int i=cur[u];i&&ans<flow;i=e[i].next) {
        cur[u]=i;
        int v=e[i].to;
        if(e[i].w&&dep[v]==dep[u]+1) {
            int x=dfs(v,min(e[i].w,flow-ans));
            if(x) e[i].w-=x,e[i^1].w+=x,ans+=x;
        }
    }
    if(ans<flow) dep[u]=-1;//说明这个点已经榨干
    return ans;
}

int main()
{
    ios::sync_with_stdio(0);
    scanf("%d %d",&n,&m);
    s=0,t=n*m+2*(n-1)*m+2*n*(m-1)+1;
    for(int i=1;i<=n;i++)for(int j=1;j<=m;j++)scanf("%d",&a),sum+=a,addedge(s,xuhao(i,j),a);
    for(int i=1;i<=n;i++)for(int j=1;j<=m;j++)scanf("%d",&a),sum+=a,addedge(xuhao(i,j),t,a);
    cnt=n*m;
    if(n-1!=0)
    {
       for(int i=1;i<=n-1;i++)
          for(int j=1;j<=m;j++)
          {
              ++cnt;
              scanf("%d",&a),sum+=a,addedge(s,cnt,a);
              addedge(cnt,xuhao(i,j),inf);
              addedge(cnt,xuhao(i+1,j),inf);
          }
       for(int i=1;i<=n-1;i++)
          for(int j=1;j<=m;j++)
          {
              ++cnt;
              scanf("%d",&a),sum+=a,addedge(cnt,t,a);
              addedge(xuhao(i,j),cnt,inf);
              addedge(xuhao(i+1,j),cnt,inf);
          }
    }
    if(m-1!=0)
    {
       for(int i=1;i<=n;i++)
          for(int j=1;j<=m-1;j++)
          {
              ++cnt;
              scanf("%d",&a),sum+=a,addedge(s,cnt,a);
              addedge(cnt,xuhao(i,j),inf);
              addedge(cnt,xuhao(i,j+1),inf);
          }
       for(int i=1;i<=n;i++)
          for(int j=1;j<=m-1;j++)
          {
              ++cnt;
              scanf("%d",&a),sum+=a,addedge(cnt,t,a);
              addedge(xuhao(i,j),cnt,inf);
              addedge(xuhao(i,j+1),cnt,inf);
          }
    }
    while(bfs())ans+=dfs(s,inf);
    printf("%d\n",sum-ans);
    return 0;
}