bzoj 1475: 方格取数 最小割

题意

在一个n*n的方格里，每个格子里都有一个正整数。从中取出若干数，使得任意两个取出的数所在格子没有公共边，且取出的数的总和尽量大。
n<=30

分析

黑白染色后最小割。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;

const int N=35;
const int inf=(int)1e9;

int n,cnt,last[N*N],s,t,dis[N*N],cur[N*N],dx[4]={0,1,0,-1},dy[4]={1,0,-1,0};
struct edge{int to,next,c;}e[N*N*N*N*2];
queue<int> que;

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

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

bool bfs()
{
    for (int i=s;i<=t;i++) dis[i]=0;
    while (!que.empty()) que.pop();
    dis[s]=1;que.push(s);
    while (!que.empty())
    {
        int u=que.front();que.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;
                que.push(e[i].to);
            }
    }
    return 0;
}

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

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

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