2022 icpc杭州站 C. No Bug No Game - 背包dp

题面

分析

能拿整个 p i p_i pi的就拿整个的,不能拿了可以拿一部分的,因此可以分成0和1两种情况,0表示拿整个的,1表示还可以拿部分的,两种情况放在一起做一遍01背包,找到最大价值。

代码
cpp 复制代码
#include <bits/stdc++.h>

#define int long long

using namespace std;

const int N = 3010;

int dp[N][N][2];
int w[N][20];
int p[N];

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int n, m;
    cin >> n >> m;
    int sum = 0;
    for(int i = 1; i <= n; i ++) {
        cin >> p[i];
        sum += p[i];
        for(int j = 1; j <= p[i]; j ++) cin >> w[i][j];
    }
    if(sum <= m) {
        int ans = 0;
        for(int i = 1; i <= n; i ++) ans += w[i][p[i]];
        cout << ans << "\n";
        return 0;
    }
    memset(dp, -0x3f, sizeof dp);
    dp[0][0][0] = 0;
    for(int i = 1; i <= n; i ++) {
        for(int j = 0; j <= m; j ++) {
            dp[i][j][0] = dp[i - 1][j][0];
            dp[i][j][1] = dp[i - 1][j][1];
            if(j >= p[i]) {
                dp[i][j][0] = max(dp[i][j][0], dp[i - 1][j - p[i]][0] + w[i][p[i]]);
                dp[i][j][1] = max(dp[i][j][1], dp[i - 1][j - p[i]][1] + w[i][p[i]]);
            }
            for(int k = 1; k < p[i]; k ++) {
                if(j >= k) dp[i][j][1] = max(dp[i][j][1], dp[i - 1][j - k][0] + w[i][k]);
            }
        }
    }
    cout << max(dp[n][m][0], dp[n][m][1]) << "\n";
}
相关推荐
我是谁??30 分钟前
C/C++使用AddressSanitizer检测内存错误
c语言·c++
小码农<^_^>32 分钟前
优选算法精品课--滑动窗口算法(一)
算法
羊小猪~~34 分钟前
神经网络基础--什么是正向传播??什么是方向传播??
人工智能·pytorch·python·深度学习·神经网络·算法·机器学习
软工菜鸡1 小时前
预训练语言模型BERT——PaddleNLP中的预训练模型
大数据·人工智能·深度学习·算法·语言模型·自然语言处理·bert
南宫生1 小时前
贪心算法习题其三【力扣】【算法学习day.20】
java·数据结构·学习·算法·leetcode·贪心算法
发霉的闲鱼1 小时前
MFC 重写了listControl类(类名为A),并把双击事件的处理函数定义在A中,主窗口如何接收表格是否被双击
c++·mfc
小c君tt1 小时前
MFC中Excel的导入以及使用步骤
c++·excel·mfc
xiaoxiao涛1 小时前
协程6 --- HOOK
c++·协程
AI视觉网奇1 小时前
sklearn 安装使用笔记
人工智能·算法·sklearn
JingHongB2 小时前
代码随想录算法训练营Day55 | 图论理论基础、深度优先搜索理论基础、卡玛网 98.所有可达路径、797. 所有可能的路径、广度优先搜索理论基础
算法·深度优先·图论