bzoj 2132: 圈地计划 网络流

题目

传送门

分析

很容易看出是用最小割来做，关键是要找出如何建图。
注意到如果不染色的话没那么好做，所以先把所有点黑板染色，s向黑点连流量为A[i,j]的边，白点向t连流量为B[i,j]的边。对于两个相邻的节点(x1,y1)和(x2,y2)，分别从x向y和y向x连一条流量为C[x1,y1]+C[x2,y2]的边，然后跑最大流即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<queue>
#define N 20005
#define inf 0x7fffffff
using namespace std;

int cnt,last[N],dis[N],dx[4]={1,0,-1,0},dy[4]={0,1,0,-1},s,t,n,m,cur[N],ans,a[105][105];
struct edge{int to,c,next,op;}e[N*100];
queue <int> q;

int point(int x,int y)
{
    return (x-1)*m+y;
}

void addedge(int u,int v,int c)
{
    e[++cnt].to=v;e[cnt].c=c;e[cnt].op=cnt+1;e[cnt].next=last[u];last[u]=cnt;
    e[++cnt].to=u;e[cnt].c=0;e[cnt].op=cnt-1;e[cnt].next=last[v];last[v]=cnt;
}

bool bfs()
{
    memset(dis,0,sizeof(dis));
    dis[s]=1;
    while (!q.empty()) q.pop();
    q.push(s);
    while (!q.empty())
    {
        int u=q.front();
        q.pop();
        for (int i=last[u];i;i=e[i].next)
            if (e[i].c&&!dis[e[i].to])
            {
                dis[e[i].to]=dis[u]+1;
                if (e[i].to==t) return 1;
                q.push(e[i].to);
            }
    }
    return 0;
}

int dfs(int x,int maxf)
{
    if (x==t||maxf==0) return maxf;
    int ret=0;
    for (int &i=cur[x];i;i=e[i].next)
        if (e[i].c&&dis[e[i].to]==dis[x]+1)
        {
            int f=dfs(e[i].to,min(maxf-ret,e[i].c));
            e[i].c-=f;
            e[e[i].op].c+=f;
            ret+=f;
            if (ret==maxf) break;
        }
    return ret;
}

void dinic()
{
    while (bfs())
    {
        for (int i=s;i<=t;i++)
            cur[i]=last[i];
        ans+=dfs(s,inf);
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    s=0;t=n*m+1;
    int sum=0;
    for (int i=1;i<=n;i++)
        for (int j=1;j<=m;j++)
        {
            int x;
            scanf("%d",&x);
            sum+=x;
            int p=point(i,j);

            if ((i+j)%2==1) addedge(s,p,x);
            else addedge(p,t,x);
        }
    for (int i=1;i<=n;i++)
        for (int j=1;j<=m;j++)
        {
            int x;
            scanf("%d",&x);
            sum+=x;
            int p=point(i,j);
            if ((i+j)%2==1) addedge(p,t,x);
            else addedge(s,p,x);
        }
    for (int i=1;i<=n;i++)
        for (int j=1;j<=m;j++)
            scanf("%d",&a[i][j]);
    for (int i=1;i<=n;i++)
        for (int j=1;j<=m;j++)
        {
            sum+=a[i][j]*4;
            if (i==1) sum-=a[i][j];if (i==n) sum-=a[i][j];
            if (j==1) sum-=a[i][j];if (j==m) sum-=a[i][j];
            int u=point(i,j);
            for (int k=0;k<4;k++)
            {
                int p=i+dx[k],q=j+dy[k];
                if (p<1||p>n||q<1||q>m) continue;
                addedge(u,point(p,q),a[i][j]+a[p][q]);
            }
        }
    dinic();
    printf("%d",sum-ans);
    return 0;
}