在一个 N×N 的矩形网格中,每个格子里都写着一个非负整数。
可以从左上角到右下角安排 K 条路线,每一步只能往下或往右,沿途经过的格子中的整数会被取走。
若多条路线重复经过一个格子,只取一次。
求能取得的整数的和最大是多少。
输入格式
第一行包含两个整数 N 和 K。
接下来 N 行,每行包含 N 个不超过 1000 的整数,用来描述整个矩形网格。
输出格式
输出一个整数,表示能取得的最大和。
数据范围
1≤N≤50,
0≤K≤10
输入样例:
3 2
1 2 3
0 2 1
1 4 2
输出样例:
15
解析:
本题我们需要在一个 n×n 的矩阵,每个格子上都写着一个非负整数,我们指定 k 条路线,每一步只能往下或往右,每个格子上的数只能取一次,求如何指定这 k 条路线,能让取得的整数和最大。
像这种方格取数问题,存在简单版本,如指定一条路线或两条路线的问题,可以使用线性 dp 来求,但是本题由于最多会有 10 条路径,用 dp 会超时,所以这里考虑用费用流来解决。
首先我们先不去考虑每个格子只能取一次的问题,我们先设法将路线转化成流。起点是左上角,终点是右下角,所有的路线都是从左上角走到右下角。所以我们可以建立一个源点和一个汇点,从源点向左上角连一条容量为 k 的边,从汇点向右下角连一条容量为 k 的边,表示有 k 条路线从左上角走到右下角。然后每个格子都可以向右和向下走,因此从每个格子向右和向下两个格子连边,且每个格子能走多次,没有限制,因此这些边的容量都是 +∞。这样我们就在网格图上建立了一个流网络。
此时可以发现流网络的任意一个可行流和原问题的任意一个方案都是一一对应的。并且原问题的每一个方案都一定有 k 条路线,所以在流网络中对应的可行流一定都是满流。这个比较直观,可以自行证明。
现在回到原题,我们要从所有路线中找出总价值最大的 k 条路线,也就是说我们要在流网络中所有的最大可行流里面找到一个费用最大的方案。那么我们就需要考虑如何将费用结合到流网络里面。
可以发现,本题的所有费用都是在格子上(点上)的,所以我们可以用拆点技巧把每个点拆成入点和出点,然后从每个入点向对应的出点连一条边,费用是该点的价值。
但是这里还有一个限制,就是每个点的价值我们都只能取走一次,以后的若干次再走到这个点上都不会在获得价值。因此我们可以对应这两种情况来建立两条边。对于每个点,从入点向出点连一条容量是 1,费用是该点价值的边,再连一条容量是 +∞,费用是 0 的边。这样我们每次走到一个点,只能走一次有费用的边来获取价值,之后的几次都只能走没有费用的边,保证每个点的价值只能取走一次。
这样本题的流网络就构建完成了,并且原问题的任意一组走法都能对应到流网络中的任意一个满流,原问题任意一组走法走过的价值之和都能对应到流网络中任意一个满流的费用。因此我们想求原问题的最大价值,等价于求流网络的最大费用最大流。
cpp
#include<iostream>
#include<string>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<utility>
#include<stack>
#include<queue>
#include<vector>
#include<set>
#include<math.h>
#include<map>
#include<sstream>
#include<deque>
#include<unordered_map>
#include<unordered_set>
#include<bitset>
using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int, int> PII;
const int N = 50*50*2+10, M = (N*4) * 2 + 10, INF = 0x3f3f3f3f;
int n, m,K, S, T;
int h[N], e[M], f[M], w[M], ne[M], idx;
int q[N], d[N], pre[N], incf[N];
bool st[N];
void add(int a, int b, int c, int d) {
e[idx] = b, f[idx] = c, w[idx] = d, ne[idx] = h[a], h[a] = idx++;
e[idx] = a, f[idx] = 0, w[idx] = -d, ne[idx] = h[b], h[b] = idx++;
}
int get(int a, int b, int c) {
return ((a - 1) * n + b)*2 + c;
}
bool spfa() {
int hh = 0, tt = 1;
memset(d, -0x3f, sizeof d);
memset(incf, 0, sizeof incf);
q[0] = S, d[S] = 0, incf[S] = 0x3f;
while (hh != tt) {
int t = q[hh++];
if (hh == N)hh = 0;
st[t] = 0;
for (int i = h[t]; i != -1; i = ne[i]) {
int j = e[i];
if (d[j] < d[t] + w[i] && f[i]) {
d[j] = d[t] + w[i];
pre[j] = i;
incf[j] = min(incf[t], f[i]);
if (!st[j]) {
st[j] = 1;
q[tt++] = j;
if (tt == N)tt = 0;
}
}
}
}
return incf[T] > 0;
}
int EK() {
int cost = 0;
while (spfa()) {
int t = incf[T];
cost += t * d[T];
for (int i = T; i != S; i = e[pre[i] ^ 1]) {
f[pre[i]] -= t;
f[pre[i] ^ 1] += t;
}
}
return cost;
}
int main() {
cin >> n >> K;
memset(h, -1, sizeof h);
S = 2 * n * n + 2, T = S + 1;
add(S, get(1, 1, 0), K, 0);
add(get(n, n, 1), T, K, 0);
for (int i = 1; i <= n; i++) {
for (int j = 1, a; j <= n; j++) {
scanf("%d", &a);
add(get(i, j, 0), get(i, j, 1), 1, a);
add(get(i, j, 0), get(i, j, 1), INF, 0);
if (i < n)add(get(i, j, 1), get(i + 1, j, 0), INF, 0);
if (j < n)add(get(i, j, 1), get(i, j + 1, 0), INF, 0);
}
}
printf("%d\n", EK());
return 0;
}