【错题集-编程题】小红取数(动态规划 - 01 背包 + 同余)

牛客对应题目链接:小红取数_牛客题霸_牛客网 (nowcoder.com)


一、分析题目

这道题是不能用空间优化的。

  • 同余原理

a % k = x 和 b % k = y <==> (a+b) % k = 0 <==> (x+y) % k = 0


状态表示

dp[i][j]:表示从前 i 个数中挑选,总和 %k 等于 j 时,最大和是多少。

  • 状态转移方程

dp[i][j] = max(dp[i-1][j], dp[i-1][(j-a[i]%k+k)%k] + a[i]);

  • 返回值

dp[n][0]


二、代码

cpp 复制代码
#include <iostream>
#include <cstring>
using namespace std; 

typedef long long LL;

const int N = 1010;

int n, k;
LL a[N];
LL dp[N][N];

int main()
{
    cin >> n >> k;
    for(int i = 1; i <= n; i++) cin >> a[i];

    memset(dp, -0x3f, sizeof dp);
    dp[0][0] = 0;

    for(int i = 1; i <= n; i++)
    {
        for(int j = 0; j < k; j++)
        {
            dp[i][j] = max(dp[i - 1][j], dp[i - 1][(j - a[i] % k + k) % k] + a[i]);
        }
    }

    if(dp[n][0] <= 0) cout << -1 << endl;
    else cout << dp[n][0] << endl;

    return 0;
}

三、反思与改进

需要熟练掌握同余定理部分的知识,虽然 01 背包会使用了,但其扩展需要通过不断提高做题量才能更好的融会贯通。

相关推荐
碧海银沙音频科技研究院3 分钟前
i2s封装成自己定义8路音频数据发送方法
arm开发·人工智能·深度学习·算法·音视频
做科研的周师兄6 分钟前
【机器学习入门】9.2:感知机的工作原理 —— 从模型结构到实战分类
人工智能·算法·机器学习·分类·数据挖掘
不去幼儿园34 分钟前
【启发式算法】狼群算法(Wolf Pack Algorithm, WPA)算法详细介绍(Python)
python·算法·启发式算法·任务分配·集群智能
墨染点香1 小时前
LeetCode 刷题【139. 单词拆分】
算法·leetcode·职场和发展
夜晚中的人海3 小时前
【C++】位运算算法习题
开发语言·c++·算法
superior tigre3 小时前
(huawei)5.最长回文子串
c++·算法
OG one.Z3 小时前
08_集成学习
人工智能·算法·机器学习
CoovallyAIHub4 小时前
超越传统3D生成:OccScene实现感知与生成的跨任务共赢
深度学习·算法·计算机视觉
Mr.H01274 小时前
克鲁斯卡尔(Kruskal)算法
数据结构·算法·图论
Tisfy4 小时前
LeetCode 3346.执行操作后元素的最高频率 I:滑动窗口(正好适合本题数据,II再另某他法)
算法·leetcode·题解·滑动窗口·哈希表