带余数动态规划|糖果

计蒜客-T1219
题解传送门

Code
记忆化搜索写起来比较方便，还比较好理解。

#include<iostream>
#include<algorithm>
#include<string.h>
#define INF 100 * 1000000;
using namespace std;

int n, k;
int val[100 + 5];
int dp[100 + 5][99];// dp[i][j]表示前i个物品中，总数%k为j的最优解

int recursion(int i, int j) {
	if (dp[i][j] != -1)
		return dp[i][j];

	if (i == 0 && j == 0)
		return dp[0][0] = 0;
	else if (i == 0)
		return dp[0][j] = -INF;// 此状态不可抵达

	dp[i][j] = max(recursion(i - 1, j), recursion(i - 1, (k + j - val[i] % k) % k) + val[i]);
	return dp[i][j];
}

int main() {
	cin >> n >> k;
	for (int i = 1; i <= n; i++) {
		cin >> val[i];
	}
	memset(dp, -1, sizeof(dp));
	int ans = recursion(n, k);
	if (ans > 0)
		cout << ans << endl;
	else
		cout << 0 << endl;// ans为负值，说明状态不可抵达，按题目要求输出0
	return 0;
}