【luogu 2744】方格取数问题

题目描述
在一个有 m*n 个方格的棋盘中，每个方格中有一个正整数。现要从方格中取数，使任意 2 个数所在方格没有公共边，且取出的数的总和最大。试设计一个满足要求的取数算法。对于给定的方格棋盘，按照取数要求编程找出总和最大的数。
输入格式：
第 1 行有 2 个正整数 m 和 n，分别表示棋盘的行数和列数。接下来的 m 行，每行有 n 个正整数，表示棋盘方格中的数。
输出格式：
程序运行结束时，将取数的最大总和输出

棋盘染色，发现如果选中一个黑点，其余的白点都不能选。
也就是，获得了黑点的价值，就不能获得白点的价值。
所以一开始我们选所有的点，看看最少舍去那些点。
从S到每个黑点连流量为点权的边，每一个黑点向四联通的点连流量正无穷的边（表示不能割），每个白点向T连流量点权的边，然后跑出最小割，用总点权-最小割就是答案。

#include<cstdio>
#include<iostream>
#include<queue>
#include<cstring>
#define S 10001
#define T 10002
using namespace std;
struct node{
    int to,next,flow,from;
}ed[1000100];
int head[10100],deep[10100],size=-1,ind[110][110],cnt=0,a[110][110],curr[10100];
int dx[]={0,0,1,-1},dy[]={1,-1,0,0};
void add(int from,int to,int flow)
{
    size++;
    ed[size].to=to;
    ed[size].from=from;
    ed[size].flow=flow;
    ed[size].next=head[from];
    head[from]=size;
}
bool bfs()
{
    queue <int> q;
    memset(deep,0,sizeof(deep));
    q.push(S);
    deep[S]=1;
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        if(u==T) return true;
        for(int i=head[u];~i;i=ed[i].next)
        {
            int v=ed[i].to;
            if(!deep[v]&&ed[i].flow)
            {
                q.push(v);
                deep[v]=deep[u]+1;
            }
        }
    }
    return false;
}
int dfs(int u,int cur)
{
    int rest=cur;
    if(u==T) return cur;
    for(int i=curr[u];~i;i=ed[i].next)
    {
        int v=ed[i].to;
        if(deep[v]==deep[u]+1&&ed[i].flow)
        {
            int new_flow=dfs(v,min(rest,ed[i].flow));
            ed[i].flow-=new_flow;
            ed[i^1].flow+=new_flow;
            rest-=new_flow;
            if(ed[i].flow) curr[u]=i;
            if(!rest) break;
        }
    }
    if(cur==rest) deep[u]=-1;
    return cur-rest;
}
int dinic()
{
    int ans=0;
    while(bfs())
    {
        for(int i=1;i<=10010;i++) curr[i]=head[i];
        ans+=dfs(S,1e9);
    }
    return ans;
}
int main()
{
    int m,n,sum=0;
    memset(head,-1,sizeof(head));
    scanf("%d%d",&m,&n);
    for(int i=1;i<=m;i++)
    {
        for(int j=1;j<=n;j++)
        {
            scanf("%d",&a[i][j]);
            ind[i][j]=++cnt;
            sum+=a[i][j];
        }
    }
    for(int i=1;i<=m;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if((i+j)%2==0)
            {
                add(S,ind[i][j],a[i][j]);
                add(ind[i][j],S,0);
                for(int k=0;k<4;k++)
                {
                    int x=i+dx[k];
                    int y=j+dy[k];
                    if(x<1||x>m||y<1||y>n) continue;
                    add(ind[i][j],ind[x][y],1e9);
                    add(ind[x][y],ind[i][j],0);
                }
            }
            else 
            {
                add(ind[i][j],T,a[i][j]);
                add(T,ind[i][j],0);
            }
        }
    }
    printf("%d",sum-dinic());
    return 0;
}