题目描述
在一个有 m×n m \times n m×n 个方格的棋盘中,每个方格中有一个正整数。
现要从方格中取数,使任意 2 2 2 个数所在方格没有公共边,且取出的数的总和最大。试设计一个满足要求的取数算法。
输入格式
文件第 1 1 1 行有 2 2 2 个正整数 m m m 和 n n n,分别表示棋盘的行数和列数。接下来的 m m m 行,每行有 n n n 个正整数,表示棋盘方格中的数。
注意:m m m 是行数,n n n 是列数。
输出格式
输出取数的最大总和。
样例
样例输入
3 3
1 2 3
3 2 3
2 3 1
样例输出
11
数据范围与提示
在一个有 m×n m \times n m×n 个方格的棋盘中,每个方格中有一个正整数。
现要从方格中取数,使任意 2 2 2 个数所在方格没有公共边,且取出的数的总和最大。试设计一个满足要求的取数算法。
输入格式
文件第 1 1 1 行有 2 2 2 个正整数 m m m 和 n n n,分别表示棋盘的行数和列数。接下来的 m m m 行,每行有 n n n 个正整数,表示棋盘方格中的数。
注意:m m m 是行数,n n n 是列数。
输出格式
输出取数的最大总和。
样例
样例输入
3 3
1 2 3
3 2 3
2 3 1
样例输出
11
数据范围与提示
1≤n,m≤30 1 \leq n, m \leq 30 1≤n,m≤30
最初想法是拆点然后跑最小费用最大流,样例都过不了。。
正解:对每个点黑白染色,保证每个点和它四周的点的颜色都不一样,源点连所有黑色点容量为1,汇点连所有白色点容量为1,所有黑色连它四周白色的点,容量为INF。
这是相当于求一个二分图最大点权独立集=总权值-最小割(定理)。
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<queue>
using namespace std;
const int maxm = 10005;
const int maxn = 100005;
const int INF = 1e9 + 7;
struct node
{
int v, flow, next;
}edge[maxm];
int dis[maxm], cur[maxm], head[maxm], f[35][35];
int n, m, s, t, sum, cnt;
void init()
{
sum = 0, cnt = 0, s = 0, t = n*m + 1;
memset(head, -1, sizeof(head));
}
void add(int u, int v, int w)
{
edge[cnt].v = v, edge[cnt].flow = w, edge[cnt].next = head[u], head[u] = cnt++;
edge[cnt].v = u, edge[cnt].flow = 0, edge[cnt].next = head[v], head[v] = cnt++;
}
int bfs()
{
queue<int>q;
memset(dis, -1, sizeof(dis));
dis[s] = 0;
q.push(s);
while (!q.empty())
{
int u = q.front();q.pop();
for (int i = head[u];i != -1;i = edge[i].next)
{
int v = edge[i].v;
if (dis[v] == -1 && edge[i].flow)
{
dis[v] = dis[u] + 1;
q.push(v);
}
}
}
if (dis[t] == -1) return 0;
return 1;
}
int dfs(int u, int flow)
{
if (u == t) return flow;
for (int i = head[u];i != -1;i = edge[i].next)
{
int v = edge[i].v;
if (dis[v] == dis[u] + 1 && edge[i].flow)
{
int d = dfs(v, min(edge[i].flow, flow));
if (d > 0)
{
edge[i].flow -= d, edge[i ^ 1].flow += d;
return d;
}
}
}
return 0;
}
int dinic()
{
int d;
while (bfs())
{
for (int i = s;i <= t;i++) cur[i] = head[i];
while (d = dfs(s, INF))
sum -= d;
}
return sum;
}
int main()
{
int i, j, k, x;
scanf("%d%d", &n, &m);
init();
for (i = 1;i <= n;i++)
for (j = 1;j <= m;j++)
scanf("%d", &f[i][j]), sum += f[i][j];
for (i = 1;i <= n;i++)
{
for (j = 1;j <= m;j++)
{
if ((i & 1 && j & 1) || (!(i & 1) && !(j & 1)))
{
add(s, (i - 1)*m + j, f[i][j]);
if (i > 1) add((i - 1)*m + j, (i - 2)*m + j, INF);
if (i < n) add((i - 1)*m + j, i*m + j, INF);
if (j > 1) add((i - 1)*m + j, (i - 1)*m + j - 1, INF);
if (j < m) add((i - 1)*m + j, (i - 1)*m + j + 1, INF);
}
else add((i - 1)*m + j, t, f[i][j]);
}
}
printf("%d\n", dinic());
return 0;
}