网络流——增广路算法（dinic）模板 [BeiJing2006]狼抓兔子

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<queue>
using namespace std;
struct data
{
    int from,to,next,cup;
    data(){from=-1,to=-1,next=-1,cup=-1;}
}e[6000001];
int cnt=0,head[1000001];
int dis[1000001];
bool vis[1000001];
void add(int u,int v,int w)
{e[cnt].from=u,e[cnt].to=v,e[cnt].next=head[u],head[u]=cnt,e[cnt].cup=w;cnt++;}
int s,t;
bool bfs()
{
    memset(dis,-1,sizeof(dis));
    int f=2147483647;
    queue<int> q;
    q.push(s);
    dis[s]=0;
    while(!q.empty())
    {
        int n=q.front();
        q.pop();
        for(int i=head[n];i>=0;i=e[i].next)
        {
            if(dis[e[i].to]==-1&&e[i].cup>0)
            {
                dis[e[i].to]=dis[n]+1;
                q.push(e[i].to);
            }
        }
    }
    return dis[t]!=-1;
}
int dfs(int n,int a)
{
    
    if(n==t||a==0) return a;
    int flow=0,f;
    for(int i=head[n];i>=0;i=e[i].next)
    {
        if(dis[e[i].to]==dis[n]+1&&e[i].cup>0)
        {
            f=dfs(e[i].to,min(a,e[i].cup));
            e[i].cup-=f;
            e[i^1].cup+=f;
            flow+=f;
            a-=f;
            if(a==0) break;
        }
    }
    if(!flow) dis[n]=-1;
    return flow;
}
int dinic()
{
    int sum=0;
    while(bfs()) sum+=dfs(s,2147483647);
    return sum;
}
int main()
{
    memset(head,-1,sizeof(head));
    int n,m;
    scanf("%d%d",&n,&m);
    int x;
    for(int i=1;i<=n;i++)
        for(int j=1;j<m;j++)
        {
            scanf("%d",&x);
            add(m*(i-1)+j,m*(i-1)+j+1,x);
            add(m*(i-1)+j+1,m*(i-1)+j,x);
        }
    for(int i=1;i<n;i++)
        for(int j=1;j<=m;j++)
        {
            scanf("%d",&x);
            add(m*(i-1)+j,m*(i)+j,x);
            add(m*(i)+j,m*(i-1)+j,x);
        }
    for(int i=1;i<n;i++)
        for(int j=1;j<m;j++)
        {
            scanf("%d",&x);
            add(m*(i-1)+j,m*(i)+j+1,x);
            add(m*(i)+j+1,m*(i-1)+j,x);
        }
    s=1,t=m*n;
    printf("%d
",dinic());
}