写在前面
思维难度不是很大的 DP,代码实现也很容易。
状态设计模式很套路,转移也很好理解。
算法思路
(因为 (k) 是常用的循环变量,下文中将题面中的模数改为 (p))
虽然要求的是模 (p) 结果为 (0) 的答案,显然这个结果受到选择的数的和可能受到模 (p) 任何余数的影响。因此不妨设计状态的时候想到用一维来表示答案模 (p) 的余数。
加上这一维之后就是一个背包模板了。
设 (f_{i, j, t, k}) 表示在第 (i) 行前 (j) 个数中选出 (t) 个数,且模 (p) 余 (k) 的最大答案。
那么显然有:
[f_{i, j, t, k} = max{f_{i, j - 1, t, k}, f_{i, j - 1, t - 1, (k - a_{i, j})mod p} + a_{i, j}}
]
当可能存在多个(或多组) (a_{i, j}) 可能将同一个答案更新的时候,取最大值即可。
这样分别处理处每一行的答案之后,再设 (g_{i, j}) 表示在前 (i) 行中选择某些数使得其和模 (p) 的余数为 (j) 的最大答案。
那么显然有:
[g_{i, j} = max{g_{i - 1, (j - k)mod p} + f_{i, m, t, k}}
]
Tips
-
注意一些边界条件。
-
处理的时候可能会出现不合法的状态,即在某一行内选出某些数,其和模 (p) 的余数根本不可能等于枚举的 (k) 的情况。这种处理方式归根到底还是因为它是从“选 (0) 个数的时候余数却不为 (0)”这个不合法状态转移来的。有个简单的处理方式,只需要将初值设为负无穷,然后将合法的初始状态(即选择数的个数和余数都为 (0))设为 (0) 就好了。
Code
-
代码中对于每一行分别处理了这一行的 (f) 数组,并紧接着更新 (g) 数组,因此将 (f) 数组删掉了一维。
-
代码中还使用了滚动数组。
#include <bits/stdc++.h>
using namespace std;
const int Maxn = 75;
namespace Read {
inline int read() {
int fh = 1, res = 0; char ch = getchar();
for(; !isdigit(ch); ch = getchar()) if(ch == '-') fh = -1;
for(; isdigit(ch); ch = getchar()) res = (res << 3) + (res << 1) + (ch ^ '0');
return fh * res;
}
}
using namespace Read;
int n, m, p;
int a[Maxn][Maxn];
int f[2][Maxn][Maxn];
int g[2][Maxn];
int main() {
n = read(); m = read(); p = read();
for(register int i = 1; i <= n; ++i) {
for(register int j = 1; j <= m; ++j) {
a[i][j] = read();
}
}
memset(g, -0x3f, sizeof(g));
g[0][0] = 0;
for(register int i = 1; i <= n; ++i) {
memset(f, -0x3f, sizeof(f));
f[0][0][0] = 0;
for(register int j = 1; j <= m; ++j) {
f[j & 1][0][0] = 0;
for(register int t = 1; t <= min(j, m/2); ++t) {
for(register int k = 0; k < p; ++k) {
f[j & 1][t][k] = max(max(f[j & 1][t][k], f[j - 1 & 1][t][k]), f[j - 1 & 1][t - 1][(k - (a[i][j] % p) + p) % p] + a[i][j]);
}
}
}
for(register int j = 0; j < p; ++j) {
g[i & 1][j] = g[i - 1 & 1][j];
for(register int t = 0; t <= (m / 2); ++t) {
for(register int k = 0; k < p; ++k) {
g[i & 1][j] = max(g[i & 1][j], g[i - 1 & 1][(j - (f[m & 1][t][k] % p) + p) % p] + f[m & 1][t][k]);
}
}
}
}
printf("%d", g[n & 1][0]);
return 0;
}