题解 | #小红取数#

小红取数

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);

}

全部评论

相关推荐

没有offer的小土豆:专业面试一般是分配面试官然后联系你面试 应该是还没给你分配对应面试官
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务