迭代改进-3（提示：费用流） 方格取数

在一个有 m×n 个方格的棋盘中，每个方格中有一个正整数。

现要从方格中取数，使任意2个数所在方格没有公共边，且取出的数的总和最大。试设计一个满足要求的取数算法。

输入格式:

文件第1行有2个正整数 m 和 n，分别表示棋盘的行数和列数。接下来的 m 行，每行有 n 个正整数，表示棋盘方格中的数。

注意：m 是行数，n 是列数。

数据保证有 1≤n,m≤30

输出格式:

输出取数的最大总和。

输入样例:

在这里给出一组输入。例如：

3 3
1 2 3
3 2 3
2 3 1

输出样例:

在这里给出相应的输出。例如：

11

代码长度限制         16 KB

时间限制                1000 ms

内存限制                64 MB

题解代码：

#include <bits/stdc++.h>
using namespace std;

const int INF = 0x3f3f3f3f; // 定义无穷大值，表示正无穷
const int MAXN = 10000; // 最大点数
const int MAXM = 10000; // 最大边数

int d[MAXN], n, m, p[MAXN], eid, S, T, a[100][100], sz, sum;

struct Edge {
    int v, c, next; // 目标节点、容量、下一条边的索引
} e[MAXM]; // 边数组

void init() {
    memset(p, -1, sizeof(p)); // 初始化邻接表头指针
    eid = 0; // 初始化边数组索引
}

void add(int u, int v, int c) {
    e[eid].v = v; // 目标节点
    e[eid].c = c; // 容量
    e[eid].next = p[u]; // 指向下一条边
    p[u] = eid++; // 更新头指针，注意是后++，表示下一个位置
}

void insert(int u, int v, int c) {
    add(u, v, c); // 添加正向边
    add(v, u, 0); // 添加反向边，容量为0
}

int bfs() {
    memset(d, -1, sizeof(d)); // 初始化距离数组为-1，表示未访问过
    queue<int> q;
    d[S] = 0; // 源点距离为0
    q.push(S); // 将源点加入队列
    while (!q.empty()) {
        int u = q.front();
        q.pop();
        for (int i = p[u]; i != -1; i = e[i].next) {
            int v = e[i].v;
            if (e[i].c > 0 && d[v] == -1) { // 容量大于0且未访问过
                d[v] = d[u] + 1; // 更新距离
                q.push(v); // 将节点加入队列
            }
        }
    }
    return (d[T] != -1); // 返回是否能够到达汇点
}

int dfs(int u, int flow) {
    if (u == T) return flow; // 到达汇点，返回流量
    int ret = 0;
    for (int i = p[u]; i != -1; i = e[i].next) {
        int v = e[i].v;
        if (e[i].c > 0 && d[v] == d[u] + 1) { // 满足条件，可以进行扩展
            int tmp = dfs(v, min(flow, e[i].c)); // 递归查找增广路径
            e[i].c -= tmp; // 更新正向边容量
            e[i ^ 1].c += tmp; // 更新反向边容量
            flow -= tmp; // 更新剩余流量
            ret += tmp; // 更新已经流过的流量
            if (!flow) break; // 如果流量为0，直接退出
        }
    }
    if (!ret) d[u] = -1; // 如果没有流过，标记为不可达
    return ret; // 返回流量
}

int Dinic() {
    int ret = 0;
    while (bfs()) { // 当存在增广路径时
        ret += dfs(S, INF); // 不断进行DFS找增广路径
    }
    return ret; // 返回总流量
}

int main() {
    init(); // 初始化
    scanf("%d %d", &m, &n); // 输入行数和列数
    S = 0; // 源点
    T = n * m + 1; // 汇点
    for (int i = 1; i <= m; i++) {
        for (int j = 1; j <= n; j++) {
            scanf("%d", &a[i][j]); // 输入每个点的权值
            sum += a[i][j]; // 计算总权值和
            sz++; // 点数加1
            if ((i + j) % 2) { // 如果是奇数层
                insert(S, sz, a[i][j]); // 连接源点到当前点
                if (j < n) insert(sz, sz + 1, INF); // 连接当前点到右边点
                if (j > 1) insert(sz, sz - 1, INF); // 连接当前点到左边点
                if (i < m) insert(sz, sz + n, INF); // 连接当前点到下面点
                if (i > 1) insert(sz, sz - n, INF); // 连接当前点到上面点
            } else { // 如果是偶数层
                insert(sz, T, a[i][j]); // 连接当前点到汇点
            }
        }
    }
    cout << sum - Dinic() << endl; // 输出最大权值和减去最大流，得到最小割的值

    return 0;
}