题解 | #小红取数#

小红取数

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

    全部评论

    相关推荐

    06-27 15:29
    门头沟学院 Java
    点赞 评论 收藏
    分享
    06-04 09:27
    门头沟学院 Java
    点赞 评论 收藏
    分享
    球Offer上岸👑:可能是大环境太差了 太卷了 学历也很重要 hc也不是很多 所以很难
    点赞 评论 收藏
    分享
    评论
    点赞
    收藏
    分享

    创作者周榜

    更多
    牛客网
    牛客网在线编程
    牛客网题解
    牛客企业服务