luogu P2774 方格取数问题 网络流24 最小割

Posted 2020-12-11 qingyuyyyyy

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#define N 2000005
using namespace std;
const int inf=1000000007;
int n,m,S,T,tmp1,tmp2,tot;
int idx,head[N],cur[N],q[N],ans1[N],ans2[N];
int e[N],ne[N],w[N];
bool b[N];
int r[N],c[N];
int ans;
bool vis[N];
int to[N];
int dp[N];
int a[N];
const int d[4][2] =  //待会枚举四个方向用的
{
    {0, 1},
    {0, -1},
    {1, 0},
    {-1, 0}
};
inline void add(int u,int v,int f)
{
    e[idx]=v;
    w[idx]=f;
    ne[idx]=head[u];
    head[u]=idx++;
}
inline bool bfs()
{
    int f=0,t=0;
    memset(cur,-1,sizeof(cur));
    q[t++]=0;
    cur[0]=0;
    while(f<t)
    {
        int now=q[f++];
        for(int i=head[now]; ~i; i=ne[i])
        {
            int v=e[i];
            if(cur[v]==-1&&w[i])
            {
                cur[v]=cur[now]+1;
                q[t++]=v;
            }
        }
    }
    if(cur[T]!=-1)
        return 1;
    return 0;
}
inline int dfs(int x,int f)
{
    if(x==T)
        return f;
    int w1,used=0;
    for(int i=head[x]; ~i; i=ne[i])
    {
        int v=e[i];
        if(cur[v]==cur[x]+1&&w[i])
        {
            w1=dfs(v,min(f-used,w[i]));
            w[i]-=w1;
            w[i^1]+=w1;
            used+=w1;
            if(used==f)
                return f;
        }
    }
    if(!used)
        cur[x]=-1;
    return used;
}
void dinic()
{
    while(bfs())
        ans+=dfs(0,0x3f3f3f3f);
}
int main()
{
    memset(head,-1,sizeof head);
    cin>>n>>m;
    S=0,T=m*n+1;
    int sum=0;
    for(int i=1; i<=n; i++)
        for(int j=1; j<=m; j++)
        {
            int w=0;
            cin>>w;
            sum+=w;
            if((i+j)%2==0)
            {
                add(S,(i-1)*m+j,w);
                add((i-1)*m+j,S,0);
                for (int k = 0; k <= 3; ++k)
                {
                    int x = i + d[k][0], y = j + d[k][1];
                    if (x>=1&&x<=n&&y>=1&&y<=m)
                    {
                        add((i-1)*m+j,(x-1)*m+y,2e9);
                        add((x-1)*m+y,(i-1)*m+j,0);
                    }
                }
            }
            else
            {
                add((i-1)*m+j, T,w);
                add(T,(i-1)*m+j,0);
            }
        }
    dinic();
    cout<<sum-ans<<endl;
    return 0;
}