题解 | #小红取数#
小红取数
https://www.nowcoder.com/practice/6a7b2b6c9e3a4f56b1db9f8ca08d889b
选择 a[i]:如果选择当前的元素 a[i],我们可以从前 i - 1 个元素中找到余数为 j 的情况。
a[i] 之后,新的余数就是 (j + a[i]) % k。dp[i][(j + a[i]) % k] 可以更新为 dp[i - 1][j] + a[i],表示我们在余数为 j 的最大和的基础上加上当前元素 a[i]不选择 a[i]:
- dp[i][(j + a[i]) % k] 维持原来的值,即 dp[i - 1][(j + a[i]) % k]。余数还是为上一个的余数
#include <algorithm>
#include <cstring>
#include <iostream>
using namespace std;
const int N = 1010;
long long a[100010];
long long dp[N][N];
int main() {
long long n, k;
cin >> n >> k;
for(int i = 1; i <= n; i ++) cin >> a[i];
for(int i = 0; i <= n; i ++)
for(int j = 0; j <= n; j ++)
dp[i][j] = -1e18;
dp[0][0] = 0;
for(int i = 1; i <= n; i ++){
for(int j = 0; j < k; j ++){
dp[i][(j + a[i]) % k] = max(dp[i - 1][j] + a[i], dp[i - 1][(j + a[i]) % k]);
}
}
if(dp[n][0] <= 0) cout << -1;
else cout << dp[n][0] << endl;
}
阿里巴巴公司氛围 652人发布
