题解 | #小红取数#
小红取数
https://www.nowcoder.com/practice/6a7b2b6c9e3a4f56b1db9f8ca08d889b
#include <iostream>
#include<vector>
using namespace std;
using i64 =long long;
int n,k;
int main()
{
ios::sync_with_stdio(0),cin.tie(0);
cin>>n>>k;
vector<i64> a(n+1);
//dp数组,dp[i][j]表示前i个数中除以k的余数为j的当前最大和
vector<vector<i64>> dp(n+1,vector<i64>(k+1,0));
for(int i=1;i<=n;i++)cin>>a[i];
for(int i=1;i<k;i++)dp[0][i]=-1;
for(int i=1;i<=n;i++)
{
for(int j=0;j<k;j++)
{
dp[i][j]=dp[i-1][j];
i64 yu = (j-a[i]%k+k)%k;
// cout<<yu<<endl;
if(dp[i-1][yu]!=-1)
{
dp[i][j]=max(dp[i][j],dp[i-1][yu]+a[i]);
}
// cout<<dp[i][j]<<" ";
}
// cout<<endl;
}
cout<<(dp[n][0]?dp[n][0]:-1);
}
