题解 | #小红取数#

小红取数

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

    全部评论

    相关推荐

    11-24 00:11
    已编辑
    广东工业大学 算法工程师
    避雷深圳&nbsp;&nbsp;yidao,试用期&nbsp;6&nbsp;个月。好嘛,试用期还没结束,就直接告诉你尽快找下一家吧,我谢谢您嘞
    牛客75408465号:笑死,直属领导和 hr 口径都没统一,各自说了一些离谱的被裁理由,你们能不能认真一点呀,哈哈哈哈哈😅😅😅
    点赞 评论 收藏
    分享
    小红书 后端选手 n*16*1.18+签字费期权
    点赞 评论 收藏
    分享
    10-24 13:36
    门头沟学院 Java
    Zzzzoooo:更新:今天下午有hr联系我去不去客户端,拒了
    点赞 评论 收藏
    分享
    点赞 收藏 评论
    分享
    牛客网
    牛客企业服务