CF1433F Zero Remainder Sum

写在前面

思维难度不是很大的 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})\bmod p} + a_{i, j}\}
\]

当可能存在多个（或多组） \(a_{i, j}\) 可能将同一个答案更新的时候，取最大值即可。

这样分别处理处每一行的答案之后，再设 \(g_{i, j}\) 表示在前 \(i\) 行中选择某些数使得其和模 \(p\) 的余数为 \(j\) 的最大答案。

那么显然有:

\[g_{i, j} = \max\{g_{i - 1, (j - k)\bmod 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;
}