糖果

发表于 更新于 分类于

Website:

  • 糖果

    Solution

  • 类似于背包，我们设f[i][j]为第i件物品，取模k为j的最大值，我们不难得出，状态转移为选或不选，不选就是
    1
    f[i][j]=f[i-1][j]

选就是

1
f[i][(f[i-1][j]+a[i])%k]=max(~,f[i-1][j]+a[i])

为了节省更多空间，我用了滚动数组，对于本题来说，数据较小，可省略。

Code

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define Maxn 110
using namespace std;
int n, k;
int a[Maxn];
int f[2][Maxn];
int main() { // g++ 糖果.cpp -o 糖果.exe -Wall 
	scanf("%d%d", &n, &k);
	for(int i = 1; i <= n; i++)
		scanf("%d", &a[i]);
	for(int i = 1; i <= n; i++) {
		for(int j = 0; j < k; j++)
			f[i & 1][j] = f[i - 1 & 1][j];
		for(int j = 0; j < k; j++)
			f[i & 1][(f[i - 1 & 1][j] + a[i]) % k] = max(f[i & 1][(f[i - 1 & 1][j] + a[i]) % k], f[i - 1 & 1][j] + a[i]);
	}
	printf("%d\n", f[n & 1][0]);
	return 0;
}

rp++